tensorflow load_weights时 Shapes (m) and shapes(255) are incompatible

网上有很多这个错误的解决办法 但是都不是我的问题,我是在用yolov3的tensorflow2.4.0的版本下,进行预测图片时,加载模型(load_weights("我的模型名字.tf").expect_partial())的时候出错,就是Shapes (x) and shapes(255) are incompatible!

主要原因是:!!我预测的时候指定的coco.names 和 训练该模型时使用的coco.names 不是同一个coco.names!那coco.names是什么文件其实也简单,就是你训练模型前有啥预定的标签,比如你要训练person,car,cat,dog等等这些标签!一行写一个!

所以,建议:训练完以后,最好把使用到的coco.names 和 模型文件(.tf) 一起保存了!因为训练前肯定要指定好标签嘛,标签就存到coco.names里面。

 

将来要预测的时候,一起下载下来,然后预测。模型文件和coco.names一定要配套使用!

不知道大家是不是这个问题导致的!

 

posted @ 2021-07-13 14:46  tianyee  阅读(222)  评论(0编辑  收藏  举报