zzh8829/yolov3-tf2

ValueError: Input 0 of layer conv2d is incompatible with the layer: : expected min_ndim=4, found ndim=3. Full shape received: 【416, 416, 3】

jiangxinufo opened this issue · 0 comments

请教下,执行迁移python train.py --dataset ./data/voc2007_train_stone.tfrecord --val_dataset ./data/voc2007_val_stone.tfrecord --classes ./data/stone.names --num_classes 1 --mode fit --transfer darknet --batch_size 4 --epochs 20 --weights ./checkpoints/yolov3.tf --weights_num_classes 80
出现如下错误是什么问题呢?如何调整程序 ValueError: Input 0 of layer conv2d is incompatible with the layer: : expected min_ndim=4, found ndim=3. Full shape received: 【416, 416, 3】