Pretrained weight model not working.

[juliussin]

I'm trying to detect some images using pretrained weight. I use yolov4-p7.pt and yolov4-p6.pt to do detection using detect.py.

The argument I use is: python detect.py --weights weights/yolov4-p7.pt --source img_dir_path --output output --img-size 1536 --conf-thres 0.001 --save-txt

But i got no result, the model did not detect anything, although all the images actually contain person and vehicles.

I have tried other weight and also without img-size and conf-thres arguments but still failed. Did I do something wrong?

[siriasadeddin]

I don't know why but it wasn't working using Cuda... I am using CPU device and started working fine.