MakeNTU-v2024
/

Image-preprocessor

Inference Endpoints

Model card Files Files and versions Community

oscarfu0501 commited on May 4

Commit

77aa4d1

•

1 Parent(s): ba28244

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -70,7 +70,7 @@ class EndpointHandler:
         else:
             x1, y1, x2, y2 = int(detections.xyxy[0][0]), int(detections.xyxy[0][1]), int(detections.xyxy[0][2]), int(detections.xyxy[0][3])
             clothes = img[y1: y2, x1: x2]
-            clothes = cv2.cvtColor(clothes, cv2.COLOR_BGR2RGB)
             retval , buffer = cv2.imencode('.jpg', clothes)
             # im_bytes = buffer.tobytes()

         else:
             x1, y1, x2, y2 = int(detections.xyxy[0][0]), int(detections.xyxy[0][1]), int(detections.xyxy[0][2]), int(detections.xyxy[0][3])
             clothes = img[y1: y2, x1: x2]
+            # clothes = cv2.cvtColor(clothes, cv2.COLOR_BGR2RGB)
             retval , buffer = cv2.imencode('.jpg', clothes)
             # im_bytes = buffer.tobytes()