Dolpheyn commited on
Commit
ec6037b
·
1 Parent(s): 7dbad54

Make detected classes appear only once

Browse files
Files changed (1) hide show
  1. model.py +2 -1
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