训练自己的数据集

[bai666ai]

课时名称	课时知识点
训练自己的数据集	训练自己的数据集

[bai666ai]

@weixin_45469554 <p>训练命令后加上--resume试试</p>

[weixin_45469554]

训练中断了，怎么用resume指令继续呢

[bai666ai]

@weixin_45469554 <p>好的，祝贺！</p>

[weixin_45469554]

训练中断问题已解决，命令是python tools\train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 1 -b 16 --fp16 -o --resume 第一次没有设置batch，默认64，造成了显存溢出，指定16正常运行

[bai666ai]

@吃零食的木偶人 <p>检查一下这个图片有没有目标，看一下标注，如果图中没有目标，可把这幅图片去掉</p> <p> </p>

[吃零食的木偶人]

File "C:\YOLOX4\yolox\evaluators\voc_eval.py", line 109, in voc_eval R = [obj for obj in recs[imagename] if obj["name"] == classname] │ │ └ 'crack' │ └ 'crack20211000' └ {'img00003': [{'name': 'ball', 'pose': 'Unspecified', 'truncated': 0, 'difficult': 0, 'bbox': [531, 447, 688, 609]}], 'img000... KeyError: 'crack20211000' 老师，每次训练自己的数据集到第10次时就会报这个错（这个是我训练集里面一张图片的名字） 这个怎么解决哦

[bai666ai]

@帅帅的青蛙 <p>安装测试图片做了吗？再检查一下文件yolox_voc_s_bm.py 是否按课件改对了</p>

[帅帅的青蛙]

请问执行训练命令： python tools/train.py -f exps/example/yolox_voc/yolox_voc_s_bm.py -d 1 -b 16 -- fp16 -o -c weights/yolox_s.pth 报错：(windows系统）怎么解决？？ ImportError:exps/example/yolox_voc/yolox_voc_s_bm.py doesn't contains class named 'Exp'

[bai666ai]

@qq_40034201 <p>你的pytorch是什么版本的？</p>

[bai666ai]

@qq_40034201 <p>你的pytorch是什么版本的？</p>

[qq_40034201]

训练nano模型的时候出错了 File"F:\PycharmProjects\YOLOX\exps/example/yolox_voc\yolox_voc_nano.py", line 90, in get_data_loader perspective=self.perspective, AttributeError: 'Exp' object has no attribute 'perspective'

[bai666ai]

@qq_40034201 <p>你的GPU型号是什么？</p>

[qq_40034201]

CUDA是10.2.89，CUDNN是7.6.5.安装都没问题，用driving.mp4调用GPU检测也可以实现。就是训练nano模型的时候会报错：ERROR | yolox.core.launch:98 - An error has been caught in function 'launch', process 'MainProcess' (2492), thread 'MainThread' (4412): AttributeError: 'Exp' object has no attribute 'perspective'。

[bai666ai]

@qq_40034201 <p>你的GPU是什么型号的？</p>

[qq_40034201]

老师，真的很需要帮助啊，可以远程指导一下么？

[bai666ai]

@qq_40034201 <p>可能是pytorch版本或cuda的问题</p>

[qq_40034201]

老师你好，训练yolox_s模型没有问题了，我把--fp16 -o去掉就可以正常训练了。但是训练nano模型的时候还是会报错 AttributeError: 'Exp' object has no attribute 'perspective'

[bai666ai]

@W-紫藤 <p>你使用的是哪个数据集训练的？先用课程中提供的数据集试试</p>

[W-紫藤]

yolox.utils.checkpoint:26 - Shape of head.cls_preds.0.weight in checkpoint is torch.Size([80, 128, 1, 1]), while shape of head.cls_preds.0.weight in model is torch.Size([2, 128, 1, 1]). 2022-03-01 16:43:04 | WARNING | yolox.utils.checkpoint:26 - Shape of head.cls_preds.0.bias in checkpoint is torch.Size([80]), while shape of head.cls_preds.0.bias in model is torch.Size([2]). 这个怎么解决，已更改coco，voc还有selfnum，还是出现问题

[bai666ai]

@我真是周周周周周 <p>目前租用GPU价格不太合适，还是推荐自己配置一个性价比高的GPU</p>

[我真是周周周周周]

老师您好,由于在本地训练一直报错,又很慢,能不能请老师录制一个在别的平台上租用GPU进行模型训练的视频,谢谢.

[bai666ai]

@i.am小超超 <p>先用课程中的数据集试一下</p>

[i.am小超超]

DataLoader worker (pid(s) 12104, 13136, 1624, 1796) exited unexpectedly 老师您好，我在训练的时候会出现这种错误，请问老师应该怎样解决

[bai666ai]

@刘臭棒Eric <p>按照项目文档做下。python肯定是要的</p>

[刘臭棒Eric]

老师，需要安装OpenCV和python么