search

WongKinYiu / PyTorch_YOLOv4

PyTorch implementation of YOLOv4
1.86k stars 585 forks source link

YoloV4-pacsp.pt #286

Open Yurushia1998 opened 3 years ago

Yurushia1998 commented 3 years ago

where can I find this file? And if possible also YoloV4-pacsp-x.pt. I try *.weights file but the performance is not as good as I expect so I want to try .pt weight

WongKinYiu commented 3 years ago

https://github.com/WongKinYiu/PyTorch_YOLOv4/tree/u5#pretrained-models--comparison

Yurushia1998 commented 3 years ago

https://github.com/WongKinYiu/PyTorch_YOLOv4/tree/u5#pretrained-models--comparison

They are all Mish models. I remember there are yoloV4 pascp normal models that does not use Mish. Do they have *.pt weight file?

WongKinYiu commented 3 years ago

yes they are, but the codebase is out of date. https://github.com/WongKinYiu/PyTorch_YOLOv4/tree/u5_preview#pretrained-models--comparison

Yurushia1998 commented 3 years ago

yes they are, but the codebase is out of date. https://github.com/WongKinYiu/PyTorch_YOLOv4/tree/u5_preview#pretrained-models--comparison

Thanks a lot !!. However, what do you when you say out of date? You means there won't be any maintenance, or there is problem in the codebase?

WongKinYiu commented 3 years ago

yes, the preview version is developed on pytorch 1.5. and current version could work with pytorch 1.7+.

Yurushia1998 commented 3 years ago

There is problems with the weights you give. When I load them, there is an error:

File "train.py", line 66, in train state_dict = {k: v for k, v in ckpt['model'].items() if model.state_dict()[k].numel() == v.numel()} File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 772, in getattr type(self).name, name)) torch.nn.modules.module.ModuleAttributeError: 'Model' object has no attribute 'items'