JoJosmin commited on
Commit
2ffe9fa
·
verified ·
1 Parent(s): d725458

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -114,10 +114,10 @@ def detect_clothing_onnx(image):
114
  # detection에서 필요한 값 추출
115
  x1, y1, x2, y2, conf, cls = detection[:6]
116
 
117
- # conf가 NumPy 배열인 경우, 단일 값으로 변환
118
  if isinstance(conf, np.ndarray):
119
- conf = conf.item() # 배열을 단일 값으로 변환
120
-
121
  if conf > 0.5: # 신뢰도 임계값 설정
122
  category = onnx_model_labels[int(cls)]
123
  categories.append({
@@ -127,6 +127,7 @@ def detect_clothing_onnx(image):
127
  })
128
 
129
  return categories
 
130
  # 이미지 자르기
131
  def crop_image(image, bbox):
132
  return image.crop((bbox[0], bbox[1], bbox[2], bbox[3]))
 
114
  # detection에서 필요한 값 추출
115
  x1, y1, x2, y2, conf, cls = detection[:6]
116
 
117
+ # conf가 배열인 경우, 최대 값을 사용
118
  if isinstance(conf, np.ndarray):
119
+ conf = np.max(conf) # 배열에서 최대 신뢰도
120
+
121
  if conf > 0.5: # 신뢰도 임계값 설정
122
  category = onnx_model_labels[int(cls)]
123
  categories.append({
 
127
  })
128
 
129
  return categories
130
+
131
  # 이미지 자르기
132
  def crop_image(image, bbox):
133
  return image.crop((bbox[0], bbox[1], bbox[2], bbox[3]))