deep_sort_pytorch icon indicating copy to clipboard operation
deep_sort_pytorch copied to clipboard

Training with my own data weight.pt The following error occurred

Open lyd959 opened this issue 4 years ago • 7 comments

Traceback (most recent call last): File "track.py", line 238, in detect(args) File "track.py", line 85, in detect model = torch.load(weights, map_location=device)['model'].float() # load to FP32 File "/home/lyd/anaconda3/envs/yolo5/lib/python3.7/site-packages/torch/serialization.py", line 592, in load return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args) File "/home/lyd/anaconda3/envs/yolo5/lib/python3.7/site-packages/torch/serialization.py", line 851, in _load result = unpickler.load() AttributeError: Can't get attribute 'C3' on <module 'models.common' from './yolov5/models/common.py'>

lyd959 avatar Apr 19 '21 10:04 lyd959

update common.py Above 4.0 can solve

lyd959 avatar Apr 19 '21 13:04 lyd959

啊兄弟,你是直接在detector.py把权重路径改成自己的的.pt嘛

Y-Mona avatar Apr 30 '21 01:04 Y-Mona

啊兄弟,你是直接在detector.py把权重路径改成自己的的.pt嘛

兄弟, 怎么用自己训练的.pth文件来运行这个跟踪程序啊.直接在detector.py里该改权重和name我试过不行,啥都不显示

tomy1998 avatar Jun 19 '21 04:06 tomy1998

麻烦问下,你们解决了么?我改了detector.py中的路径,还有yaml文件中的路径,还利用 @Y-Mona 您的方法把权重转化了,但还是运行跟踪时检测不到目标,麻烦您可以解答一下吗?@18795908697

啊兄弟,你是直接在detector.py把权重路径改成自己的的.pt嘛

兄弟, 怎么用自己训练的.pth文件来运行这个跟踪程序啊.直接在detector.py里该改权重和name我试过不行,啥都不显示

yuanqiyin avatar Jul 15 '21 07:07 yuanqiyin

啊兄弟,你是直接在detector.py把权重路径改成自己的的.pt嘛

兄弟, 怎么用自己训练的.pth文件来运行这个跟踪程序啊.直接在detector.py里该改权重和name我试过不行,啥都不显示

还有别的文件要改,配置的.yaml什么的,可以进跟踪执行文件里看一下,需要权重的都改

Y-Mona avatar Jul 20 '21 00:07 Y-Mona

啊兄弟,你是直接在detector.py把权重路径改成自己的的.pt嘛

兄弟, 怎么用自己训练的.pth文件来运行这个跟踪程序啊.直接在detector.py里该改权重和name我试过不行,啥都不显示

还有别的文件要改,配置的.yaml什么的,可以进跟踪执行文件里看一下,需要权重的都改

您好,我改了所有用到权重文件的地方,还是得不到跟踪的目标,用代码中带的预训练权重和对应的names文件是可以的,请问我是模型或者是哪里出现了问题么? 我暂时做了以下工作: 1.用自己的pytorch版本的yolov3训练,然后得到yolov3.pt文件; 2.利用自己训练的cfg文件进行转化,得到yolov3.weights文件; 3.替换deepsort中所有用到检测权重的地方,例如yolov3.yaml和deepsort.yaml文件; 烦请您看到回复一下,谢谢您。

yuanqiyin avatar Jul 22 '21 03:07 yuanqiyin

你好,我遇到了同样的问题。请问你是怎么解决的?

ASCII-cxy avatar Nov 16 '22 12:11 ASCII-cxy