Open saikrishnadas opened 4 years ago
yes, yolov4.pt should also work.
Follow these steps https://github.com/wang-xinyu/tensorrtx/tree/master/yolov4#excute, not using this repo, but using https://github.com/ultralytics/yolov3
converting darknet to yolov4 works fine. But its too slow to run on my jetson nx (took about 64 secs for detection ). I need to convert the yolov4.pt file into tensorrt, so that the model runs faster
Follow these steps https://github.com/wang-xinyu/tensorrtx/tree/master/yolov4#excute, to convert yolov4.pt file into tensorrt
Doesn't that convert yolov4.weight to yolov4.wts and to .engine? I have weight named yolov4.pt
gen_wts.py can also convert yolov4.pt to yolov4.wts. Can you try it ?
@wang-xinyu NameError : name 'device' is not defined. in model.load_state_dict(torch.load(weights,map_location=device)['model'])
I have a yolov4.weight file converted to pytorch, and i would like to know if this repo can convert the yolov4.pt to tensorrt ?