yolov5_v4，模型weight转engine的时候崩溃

enazoe / yolo-tensorrt

TensorRT8.Support Yolov5n,s,m,l,x .darknet -> tensorrt. Yolov4 Yolov3 use raw darknet *.weights and *.cfg fils. If the wrapper is useful to you,please Star it.

MIT License

1.19k stars 315 forks source link

yolov5_v4，模型weight转engine的时候崩溃 #129

Closed xinsuinizhuan closed 3 years ago

xinsuinizhuan commented 3 years ago

windows10, cuda_11.0 cuda_11.0.2_451.48_win10 cudnn cudnn-11.2-windows-x64-v8.1.0.77 tensorrt TensorRT-7.2.3.4.Windows10.x86_64.cuda-11.1.cudnn8.1 vs2019 https://github.com/ultralytics/yolov5/tree/v4.0 下载的模型第一步，执行yaml2cfg.py,生成的cfg和weight拷贝到config下第二步，执行程序，崩溃