Make detected classes appear only once
Browse files
model.py
CHANGED
@@ -16,6 +16,7 @@ class yolox():
|
|
16 |
self.confThreshold = confThreshold
|
17 |
self.nmsThreshold = nmsThreshold
|
18 |
self.objThreshold = objThreshold
|
|
|
19 |
def preprocess(self, image):
|
20 |
if len(image.shape) == 3:
|
21 |
padded_img = np.ones((self.input_size[0], self.input_size[1], 3)) * 114.0
|
@@ -149,5 +150,5 @@ class yolox():
|
|
149 |
if dets is not None:
|
150 |
final_boxes, final_scores, final_cls_inds = dets[:, :4], dets[:, 4], dets[:, 5]
|
151 |
srcimg, detected_classes = self.vis(srcimg, final_boxes, final_scores, final_cls_inds)
|
152 |
-
return srcimg, ", ".join(detected_classes) if len(detected_classes) > 0 else ""
|
153 |
|
|
|
16 |
self.confThreshold = confThreshold
|
17 |
self.nmsThreshold = nmsThreshold
|
18 |
self.objThreshold = objThreshold
|
19 |
+
|
20 |
def preprocess(self, image):
|
21 |
if len(image.shape) == 3:
|
22 |
padded_img = np.ones((self.input_size[0], self.input_size[1], 3)) * 114.0
|
|
|
150 |
if dets is not None:
|
151 |
final_boxes, final_scores, final_cls_inds = dets[:, :4], dets[:, 4], dets[:, 5]
|
152 |
srcimg, detected_classes = self.vis(srcimg, final_boxes, final_scores, final_cls_inds)
|
153 |
+
return srcimg, ", ".join(list(set(detected_classes))) if len(detected_classes) > 0 else ""
|
154 |
|