Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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가
|
118 |
if isinstance(conf, np.ndarray):
|
119 |
-
conf =
|
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]))
|