отслеживание 2 объектов и предоставление им разных ограничивающих рамок - PullRequest
Новая
       43

отслеживание 2 объектов и предоставление им разных ограничивающих рамок

0 голосов
/ 16 апреля 2020

Я пытаюсь адаптировать этот код к моему объекту в отслеживании и обнаружении движущегося объекта в камере наблюдения. Я пытаюсь работать только на двух классах автомобилей и мотоциклов, но каждый раз, когда я пытаюсь получить ошибку, это код, который я использую зеленые цвета красный для ограничительной рамки 

cmap = plt.get_cmap('tab20b')
colors = [41,42]


for track in tracker.tracks:
            if class_name == "car" :
              if not track.is_confirmed() or track.time_since_update > 1 :
                continue 
                bbox = track.to_tlbr()
                class_name = track.get_class()
                color = 42
                cv2.rectangle(img, (int(bbox[0]), int(bbox[1])), (int(bbox[2]), int(bbox[3])), color, 2)
                cv2.rectangle(img, (int(bbox[0]), int(bbox[1]-30)), (int(bbox[0])+(len(class_name)+len(str(track.track_id)))*17, int(bbox[1])), color, -1)
                cv2.putText(img, class_name +  "-" + str(track.track_id),(int(bbox[0]), int(bbox[1]-10)),0, 0.75, (255,255,255),2)
            elif class_name == "motocycle" :
              if not track.is_confirmed() or track.time_since_update > 1 :
                continue 
                bbox = track.to_tlbr()
                class_name = track.get_class()
                color = 41
                cv2.rectangle(img, (int(bbox[0]), int(bbox[1])), (int(bbox[2]), int(bbox[3])), color, 2)
                cv2.rectangle(img, (int(bbox[0]), int(bbox[1]-30)), (int(bbox[0])+(len(class_name)+len(str(track.track_id)))*17, int(bbox[1])), color, -1)
                cv2.putText(img, class_name + "-" + str(track.track_id),(int(bbox[0]), int(bbox[1]-10)),0, 0.75, (255,255,255),2)
...