cv-detect-robot

cv-detect-robot copied to clipboard

yolov5哪个版本

1

这个pt模型是用哪个版本训练的，我试了v5.0和v3.0都不行。V5可以转wts,但是engine出问题，V3都转不了wts Traceback (most recent call last): File "gen_wts.py", line 8, in model = torch.load('weights/number_v5s.pt', map_location=device)['model'].float() # load to FP32 File "/home/lpx/.local/lib/python3.8/site-packages/torch/serialization.py", line 789, in load return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)...

teacher1314