pt文件转换问题 about efficientteacher HOT 9 OPEN

您好，请问您开始训练的时候不会报找不到target里的图片标签的错误吗？请问您的数据集格式是什么样子的

from efficientteacher.

我在训练好efficient_yolol.pt之后，使用convert_efficient_to_yolov5转换成标准的yolov5l.pt，但在使用yolov5官方的detect.py时，出现了以下bug： Traceback (most recent call last): File "/home/zzz/Desktop/workspace/yolov5/detect.py", line 307, in main(opt) File "/home/zzz/Desktop/workspace/yolov5/detect.py", line 302, in main run(* * vars(opt)) File "/usr/local/anaconda3/envs/***/lib/python3.8/site-packages/torch/autograd/grad_mode.py", line 28, in decorate_context return func(*args, * * kwargs) File "/home/zzz/Desktop/workspace/yolov5/detect.py", line 82, in run model = torch.jit.load(w) if 'torchscript' in w else attempt_load(weights, map_location=device) File "/home/zzz/Desktop/workspace/yolov5/models/experimental.py", line 94, in attempt_load ckpt = torch.load(attempt_download(w), map_location=map_location) # load File "/usr/local/anaconda3/envs/yolov5/lib/python3.8/site-packages/torch/serialization.py", line 607, in load return _load(opened_zipfile, map_location, pickle_module, * * pickle_load_args) File "/usr/local/anaconda3/envs/yolov5/lib/python3.8/site-packages/torch/serialization.py", line 882, in _load result = unpickler.load() File "/usr/local/anaconda3/envs/yolov5/lib/python3.8/site-packages/torch/serialization.py", line 875, in find_class return super().find_class(mod_name, name) AttributeError: Can't get attribute 'DetectionModel' on <module 'models.yolo' from '/home/zzz/Desktop/workspace/yolov5/models/yolo.py'>

您好，请问您开始训练的时候不会报找不到target里的图片标签的错误吗？请问您的数据集格式是什么样子的

from efficientteacher.

您好，请问您开始训练的时候不会报找不到target里的图片标签的错误吗？请问您的数据集格式是什么样子的

把您的配置数据的yaml发我看一下。

from efficientteacher.

您好，请问您开始训练的时候不会报找不到target里的图片标签的错误吗？请问您的数据集格式是什么样子的

把您的配置数据的yaml发我看一下。

from efficientteacher.

您好，请问您开始训练的时候不会报找不到target里的图片标签的错误吗？请问您的数据集格式是什么样子的

把您的配置数据的yaml发我看一下。

你把burn_epochs设置成0试试

from efficientteacher.

您好，请问您开始训练的时候不会报找不到target里的图片标签的错误吗？请问您的数据集格式是什么样子的

你的训练指令是什么？

from efficientteacher.

您好，请问您开始训练的时候不会报找不到target里的图片标签的错误吗？请问您的数据集格式是什么样子的

你的训练指令是什么？

from efficientteacher.

您好，请问您开始训练的时候不会报找不到target里的图片标签的错误吗？请问您的数据集格式是什么样子的

你的训练指令是什么？

burn_epochs设置0也没有作用

ssod_hyp的with_gt应该设置成False，你设置成了True，应该是默认你的unlabel数据也是有gt的。
顺便问一句，阿里只给了yolov5l的custom.yaml，你是仿照哪个文件写的yolov5s.yaml的

from efficientteacher.

您好，请问您开始训练的时候不会报找不到target里的图片标签的错误吗？请问您的数据集格式是什么样子的

你的训练指令是什么？

burn_epochs设置0也没有作用

ssod_hyp的with_gt应该设置成False，你设置成了True，应该是默认你的unlabel数据也是有gt的。 顺便问一句，阿里只给了yolov5l的custom.yaml，你是仿照哪个文件写的yolov5s.yaml的

作者工程代码里面的yaml文件很多，按照readme上面的指示我并没有找到对应目录下的yaml文件，不知道是不是作者忘记更新readme了，我是参考这个目录下的yaml写的configs/ssod/custom/yolov5l_custom_ssod.yaml
顺便问一下，你找到了作者readme里面说明的那个yaml文件了吗？

from efficientteacher.