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...

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