yolov7
yolov7 copied to clipboard
Error when inference on video
I use docker image nvcr.io/nvidia/pytorch:22.08-py3 for run yolov7
I can easily inference on image by:
python detect.py --weights yolov7.pt --conf 0.25 --img-size 640 --source inference/images/horses.jpg
but when inference on video by:
detect.py --weights yolov7.pt --conf 0.25 --img-size 640 --source yourvideo.mp4
I got this error:
/opt/conda/lib/python3.8/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 /opt/pytorch/pytorch/aten/src/ATen/native/TensorShape.cpp:2156.)
return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]
Done. (0.000s)
But have no video output How can fix it ?
Thanks for reading and hope get reply from everyone ! Thanks you so much !