YOLOv5_PyTorch_cpp

YOLOv5_PyTorch_cpp copied to clipboard

after successful build running this command yields error while loading shared libraries: libopencv_dnn.so.4.4: ./main --input-dir ../../input_image/ --model-file ../../model/yolov5s_torchscript_B1-C3-H640-W640_torch1-6-0_cuda10-2_cpu.pt I am not sure what am I doing wrong

razor123max

TongShuo-Demon

first and second test speed very slow

5

Hello, my cuda10.2 libtorch1.7.1 model yolov5 was slow in the first two tests and the second one was the slowest and then normal. What is the reason?  

WANGCHAO1996

You've tried infer with many batch ?

2

zhongqingyang

模型导出

2

请问在导出gpu版本的模型的时候，设置的model.model[-1].export = False吗？

zenith779