代码问题

[YueYiHuan]

您好，首先很感谢您能够开源代码，我用您的工程在自己的数据集上测试了下，出现了两个问题： 1、load_weights时会出现维度不匹配的问题，在load_darknet_weights函数中注释“”“自己加的”那段代码后即可运行； 2、dataset.py中有一个labels4.append(labels)应该放在if内。

[coldlarry]

非常感谢您的回复！我会查看后尽快修正。

------------------ 原始邮件 ------------------ 发件人: "YueYiHuan"<[email protected]>; 发送时间: 2019年11月5日(星期二) 上午10:42 收件人: "coldlarry/YOLOv3-complete-pruning"<[email protected]>; 抄送: "Subscribed"<[email protected]>; 主题: [coldlarry/YOLOv3-complete-pruning] 代码问题 (#2)

您好，首先很感谢您能够开源代码，我用您的工程在自己的数据集上测试了下，出现了两个问题： 1、load_weights时会出现维度不匹配的问题，在load_darknet_weights函数中注释“”“自己加的”那段代码后即可运行； 2、dataset.py中有一个labels4.append(labels)应该放在if内。

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or unsubscribe.

[YueYiHuan]

非常感谢您的回复！我会查看后尽快修正。 … ------------------ 原始邮件 ------------------ 发件人: "YueYiHuan"<[email protected]>; 发送时间: 2019年11月5日(星期二) 上午10:42 收件人: "coldlarry/YOLOv3-complete-pruning"<[email protected]>; 抄送: "Subscribed"<[email protected]>; 主题: [coldlarry/YOLOv3-complete-pruning] 代码问题 (#2) 您好，首先很感谢您能够开源代码，我用您的工程在自己的数据集上测试了下，出现了两个问题： 1、load_weights时会出现维度不匹配的问题，在load_darknet_weights函数中注释“”“自己加的”那段代码后即可运行； 2、dataset.py中有一个labels4.append(labels)应该放在if内。 — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or unsubscribe.

对了，稀疏训练过程中的MAP一直显示是0，目前还没找出问题所在。

[coldlarry]

您好，您说的问题我遇到过，但只是训练过程中的前几轮会出现这个情况。我觉得可能是如下原因：①载入权重的时候，载入有问题，导致模型初始化有误，所以训练过程map一直为0. ②训练时用的稀疏化因子太大？比如使用0.01的话，map就会接近0 对于问题①，我今天下午更新了models.py，您看下可以跑得通吗？ 我在hand数据集上，一切都正常。

[YueYiHuan]

您好，您说的问题我遇到过，但只是训练过程中的前几轮会出现这个情况。我觉得可能是如下原因：①载入权重的时候，载入有问题，导致模型初始化有误，所以训练过程map一直为0. ②训练时用的稀疏化因子太大？比如使用0.01的话，map就会接近0 对于问题①，我今天下午更新了models.py，您看下可以跑得通吗？ 我在hand数据集上，一切都正常。

因为我之前是在Lam1360的工程上做过剪枝，当时设置的稀疏化因子是0.005，稀疏训练时没有问题的，至于您说的第一个原因我还没有测试，测试过之后我再回复您！

[coldlarry]

建议先设置为0.001

[yuanlei-2018]

您好，首先很感谢您能够开源代码，我用您的工程在自己的数据集上测试了下，出现了两个问题： 1、load_weights时会出现维度不匹配的问题，在load_darknet_weights函数中注释“”“自己加的”那段代码后即可运行； 2、dataset.py中有一个labels4.append(labels)应该放在if内。

我也出现了同样的问题：RuntimeError: shape '[256, 128, 3, 3]' is invalid for input of size 246683 代码那一段注释后 File "train.py", line 479, in train() # train normally File "train.py", line 134, in train cutoff = load_darknet_weights(model, weights) File "/home/yuanlei/YOLOv3-complete-pruning-master/models.py", line 394, in load_darknet_weights assert ptr == len(weights)
AssertionError

[yuanlei-2018]

您好，首先很感谢您能够开源代码，我用您的工程在自己的数据集上测试了下，出现了两个问题： 1、load_weights时会出现维度不匹配的问题，在load_darknet_weights函数中注释“”“自己加的”那段代码后即可运行； 2、dataset.py中有一个labels4.append(labels)应该放在if内。

希望帮忙解决