IndexError: list index out of range

gpqls7669 · May 6, 2020, 12:17pm

Description

Traceback (most recent call last):
File “trt_ssd.py”, line 102, in
main()
File “trt_ssd.py”, line 94, in main
loop_and_detect(cam, trt_ssd, conf_th=0.3, vis=vis)
File “trt_ssd.py”, line 64, in loop_and_detect
img = vis.draw_bboxes(img, boxes, confs, clss)
File “/home/hyebin/project/tensorrt_demos/utils/visualization.py”, line 96, in draw_bboxes
color = self.colors[cl]
IndexError: list index out of range

I got this error, then I got this answer from jk-Jung.

You need to initialize “BBoxVisualization” with the correct “cls_dict” (class dictionary, which contains the correct number of classes).

github.com

jkjung-avt/tensorrt_demos/blob/master/trt_ssd.py#L92


      
          def main():
              args = parse_args()
              cam = Camera(args)
              if not cam.isOpened():
                  raise SystemExit('ERROR: failed to open camera!')
          
              cls_dict = get_cls_dict(args.model.split('_')[-1])
              trt_ssd = TrtSSD(args.model, INPUT_HW)
          
              open_window(
                  WINDOW_NAME, 'Camera TensorRT SSD Demo',
                  cam.img_width, cam.img_height)
              vis = BBoxVisualization(cls_dict)
              loop_and_detect(cam, trt_ssd, conf_th=0.3, vis=vis)
          
              cam.release()
              cv2.destroyAllWindows()
          
          
          if __name__ == '__main__':
              main()

My number of class is 1.
I don’t know what he is saying.
How to modify the code?

Environment

SunilJB · May 6, 2020, 6:12pm

Hi,

Please refer to below topic, in case it helps:

Thanks

Virat_Baahubali · January 2, 2021, 10:18pm

Gtk-Message: 16:50:34.237: Failed to load module "canberra-gtk-module"
Traceback (most recent call last):
  File "camera_tf_trt.py", line 198, in <module>
    main()
  File "camera_tf_trt.py", line 188, in main
    loop_and_detect(cam, tf_sess, args.conf_th, vis, od_type=od_type)
  File "camera_tf_trt.py", line 120, in loop_and_detect
    img = vis.draw_bboxes(img, box, conf, cls)
  File "/home/shlok/projects/tf_trt_models/utils/visualization.py", line 96, in draw_bboxes
    color = self.colors[cl]
IndexError: list index out of range

I am running GitHub - jkjung-avt/tf_trt_models: TensorFlow object detection models accelerated with NVIDIA TensorRT (TF-TRT) using this command
python3 camera_tf_trt.py --latency 200 --model new_model_ssd_colab_v2_mobilenet_2020_29_12 --labelmap data/label_final.pbtxt --num-classes 3

I have only 3 classes in label file(bottle,cardboard,clothing). I am running my custom model was able to create tf-trt graph of my model successfully but as soon as I run it using onboard camera it freezes and gives index out of range errror

{0: 'CLS0', 1: 'bottle', 2: 'cardboard', 3: 'clothing'} this is the list of the classes in cls_dict so I changed --num-classes 4 still it didn’t work. I am not sure about 0: 'CLS0' is that for background? I can see this class gets added here along with the classes from label file /home/s/projects/tf_trt_models/utils/od_utils.py

def read_label_map(path_to_labels):
    """Read from the label map file and return a class dictionary which
    maps class id (int) to the corresponding display name (string).

    Reference:
    https://github.com/tensorflow/models/blob/master/research/object_detection/object_detection_tutorial.ipynb
    """
    from object_detection.utils import label_map_util

    category_index = label_map_util.create_category_index_from_labelmap(
        path_to_labels)
    cls_dict = {int(x['id']): x['name'] for x in category_index.values()}
    num_classes = max(c for c in cls_dict.keys()) + 1
    # add missing classes as, say,'CLS12' if any
    return {i: cls_dict.get(i, 'CLS{}'.format(i)) for i in range(num_classes)}

Can anyone guide me how should I run it on jetson tx2 with onboard cam?
thanks