本文共 899 字,大约阅读时间需要 2 分钟。
在jetsonnano ubuntu18.04上进行测试,在传了文件过去之后一直报错,尤其是这个 filters = int(module_def[‘filters’])
ValueError: invalid literal for int() with base 10: ‘’一直以为是代码哪出了问题,看了半天没看出来,然后查也没查到。
另外好像notepadqq不支持arm64,所以尝试了好一会一直没法下成功…无奈啊 最终的结论就是:不是代码的报错,而是自己的权重文件cfg出错,是因为没有传正确的cfg所导致, 自定义权重是yolov3-custom.cfg且权重问题它还会造成这个报错:
self.class.name, “\n\t”.join(error_msgs))) RuntimeError: Error(s) in loading state_dict for Darknet: Missing key(s) in state_dict: “module_list.1.conv_1.weight”, “module_list.1.batch_norm_1.weight”, “module_list.1.batch_norm_1.bias”, “module_list.1.batch_norm_1.running_mean”, “module_list.1.batch_norm_1.running_var”, "module_list.3 size mismatch for module_list.0.batch_norm_0.running_var: copying a param with shape torch.Size([16]) from checkpoint, the shape in current model is torch.Size([32]). 看起来是图片尺寸不匹配,其实还是参数的设置有问题。所以对于有的报错,还真不能只看上面的提示,不然永远找不到问题。 跑通之后: 已经有检测框结果了。提醒自己,配置文件很重要,细节不可忽略!!!
转载地址:http://avten.baihongyu.com/