TypeError: 'float' object is not subscriptable

[RioChan]

Traceback (most recent call last):
File "detect.py", line 238, in
main(opt)
File "detect.py", line 233, in main
run(**vars(opt))
File "/root/anaconda3/lib/python3.6/site-packages/torch/autograd/grad_mode.py", line 28, in decorate_context
return func(*args, **kwargs)
File "detect.py", line 141, in run
det[:, :4] = scale_coords(img.shape[2:], det[:, :4], im0.shape).round()
File "/opt/edgeai-yolov5/utils/general.py", line 471, in scale_coords
coords[:, [0, 2]] /= gain[1]
TypeError: 'float' object is not subscriptable

May you help to solve this error?
I'm not sure if it can be solved by this way in the ./utils/general.py:

coords[:, [0, 2]] /= gain[1]
coords[:, [1, 3]] /= gain[0]

coords[:, [0, 2]] /= gain

coords[:, [1, 3]] /= gain

[debapriyamaji]

Can u please share the steps to reproduce the error?

[RioChan]

Hi Maji:

I just issue this cmd below and it gives me this error:

root@ubuntu-vm:/opt/edgeai-yolov5# python detect.py --source https://www.youtube.com/watch?v=mMp4UDNII6Q

Traceback (most recent call last):
File "detect.py", line 238, in
main(opt)
File "detect.py", line 233, in main
run(**vars(opt))
File "/root/anaconda3/lib/python3.6/site-packages/torch/autograd/grad_mode.py", line 28, in decorate_context
return func(*args, **kwargs)
File "detect.py", line 141, in run
det[:, :4] = scale_coords(img.shape[2:], det[:, :4], im0.shape).round()
File "/opt/edgeai-yolov5/utils/general.py", line 471, in scale_coords
coords[:, [0, 2]] /= gain[1]
TypeError: 'float' object is not subscriptable

BR Rio

[RioChan]

Even I issue this cmd, it gives me the same error:

python detect.py --source ./data/images/ --weights yolov5s.pt --conf 0.4

[RioChan]

If I made this modification, it seems working okay.
please confirm.

#Original

coords[:, [0, 2]] /= gain[1]

coords[:, [1, 3]] /= gain[0]

#Rio, might fixed, TBC.
coords[:, [0, 2]] /= gain
coords[:, [1, 3]] /= gain

is this Working okay log?

root@ubuntu-vm:/opt/edgeai-yolov5# python detect.py --source ./data/images/ --weights yolov5s.pt --conf 0.4
detect: weights=['yolov5s.pt'], source=./data/images/, imgsz=640, conf_thres=0.4, iou_thres=0.45, max_det=1000, device=, view_img=False, save_txt=False, save_conf=False, save_crop=False, nosave=False, classes=None, agnostic_nms=False, augment=False, visualize=False, update=False, project=runs/detect, name=exp, exist_ok=False, line_thickness=3, hide_labels=False, hide_conf=False, half=False
YOLOv5 🚀 v5.0-324-gca6d78f torch 1.10.1+cu102 CPU

Fusing layers...
/root/anaconda3/lib/python3.6/site-packages/torch/functional.py:445: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:2157.)
return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]
Model Summary: 213 layers, 7225885 parameters, 0 gradients, 16.5 GFLOPs
image 1/2 /opt/edgeai-yolov5/data/images/bus.jpg: 640x480 3 persons, 1 bus, Done. (0.162s)
image 2/2 /opt/edgeai-yolov5/data/images/zidane.jpg: 384x640 2 persons, 1 tie, Done. (0.107s)
Results saved to runs/detect/exp6
Done. (0.579s)

[RioChan]

Hi

Any comments / Feedback?

BR Rio

[debapriyamaji]

Hi Rio,
This looks fine. Thanks for finding it out. Will fix it.

Regards, Debapriya

[RioChan]

Debapriya, thanks, thumb up !