Bil369 / MaskDetect-YOLOv4-PyTorch

基于PyTorch&YOLOv4实现的口罩佩戴检测 :star: 自建口罩数据集分享
GNU General Public License v3.0
203 stars 53 forks source link

公开模型加载有问题 #4

Open congmonkey opened 3 years ago

congmonkey commented 3 years ago

你好,我这边下载百度云的yolov4_maskdetect_weights1.pth在torch.load()的时候报错了,可能是模型有点问题, 请问还有其他模型下载路径吗。

Bil369 commented 3 years ago

你好,我这边下载百度云的yolov4_maskdetect_weights1.pth在torch.load()的时候报错了,可能是模型有点问题, 请问还有其他模型下载路径吗。

你好,是不是模型路径设置错误呢(看一看train.ipynb中的model_path),我这边测试过是可以正常加载的。