YuHengsss / YOLOV

This repo is an implementation of PyTorch version YOLOV Series
Apache License 2.0
278 stars 39 forks source link

Error in online video demo #36

Closed ayushag-glitch closed 1 year ago

ayushag-glitch commented 1 year ago

Getting the following error while running the online demo script:

Traceback (most recent call last):
  File "tools/yolov_demo_online.py", line 320, in <module>
    main(exp, args)
  File "tools/yolov_demo_online.py", line 313, in main
    imageflow_demo(predictor, vis_folder, current_time, args)
  File "tools/yolov_demo_online.py", line 226, in imageflow_demo
    N = int(res_dict['cls_scores'].shape[0] / len(tmp_imgs))
TypeError: list indices must be integers or slices, not str

I have not changed anything in the script. Although I did check that pred_result, and res_dict are equal which are obtained from the inference step.

pred_result, res_dict = predictor.inference(imgs, other_result)

Can someone please guide me