错误描述
用TF2.2版本训练自己的SSD模型,出现shape维度不匹配的问题。‘conv4_3_norm_mbox_loc’ shape mismatch
原因:
自己的数据集不是VOC2007的20类,而是5类,使用预训练模型**ssd_weight.py,**就会出现这个情况,需要重新训练模型.,生成自己的权重。运行 train.py
%load train.py
然后再运行ssd.py
注意修改类别数,还有权重。model_path和classes_path的路径。这里路径如果出现错误,请加载绝对路径。
class SSD(object):
_defaults = {
"model_path" : 'model_data/best_weight.h5',
"classes_path" : 'model_data/voc_classes.txt',
"input_shape" : (300, 300, 3),
"confidence" : 0.9,
"nms_iou" : 0.45,
'anchors_size' : [30,60,111,162,213,264,315],
最后在测试,运行predict.py!
完成!!!