search

zylo117 / Yet-Another-EfficientDet-Pytorch

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.
GNU Lesser General Public License v3.0
5.2k stars 1.27k forks source link

从头训练数据集问题 #228

Open yemu1138178251 opened 4 years ago

yemu1138178251 commented 4 years ago

我按照步骤做的时候发生了报错 error: (-215:Assertion failed) !_src.empty() in function 'cv::cvtColor' 经查在图片路径上发现了问题 但是没有找到如何修改 图片路径打印出来是 datasets/defect\train.PNG 望帮忙解决 我是新手。。

zylo117 commented 4 years ago

training on Windows? filepath issues.

try to use / instead of \.

gtgtgt1117 commented 4 years ago

我按照步骤做的时候发生了报错 error: (-215:Assertion failed) !_src.empty() in function 'cv::cvtColor' 经查在图片路径上发现了问题 但是没有找到如何修改 图片路径打印出来是 datasets/defect\train.PNG 望帮忙解决 我是新手。。

图片路径有问题,打印一下图片path,根据数据所在位置修改就好了

zhouzhubin commented 4 years ago

@zylo117 您好，我也训练了自己的数据，训练出来，测试了一下，发现box与标注框有点差异。我的训练集有1000张，训练了100个epcho，loss基本不怎么下降了，才停下的。我应该调哪些超参啊？

zylo117 commented 4 years ago

run evaluation. check for overfitting

123wk45678 commented 4 years ago

我想问一下，你跑自己的训练集，训练出模型之后，你test成功了吗？我训练好模型之后，想test自己的模型效果，发现我的模型test不了，好像是因为backbone.py的原因，问一下你遇到这种问题了吗？

123wk45678 commented 4 years ago

RuntimeError: Error(s) in loading state_dict for EfficientDetBackbone: Missing key(s) in state_dict: "bifpn.4.p6_w1", "bifpn.4.p5_w1", "bifpn.4.p4_w1", "bifpn.4.p3_w1", "bifpn.4.p4_w2", "bifpn.4.p5_w2", "bifpn.4.p6

我报的是这种问题，好像是我训练的模型torch.szie()不对吗？

zhouzhubin commented 4 years ago

我好像没有报这种错误,你改了train.py吗?我就是直接用大佬的代码,好像就没改

jun214384468 commented 3 years ago

我想问一下，你跑自己的训练集，训练出模型之后，你test成功了吗？我训练好模型之后，想test自己的模型效果，发现我的模型test不了，好像是因为backbone.py的原因，问一下你遇到这种问题了吗？

你好，我也遇到了同样的错误，请问你是怎么解决的呢