使用自己数据集训练yolov3增强模型时报错
Created by: llzhaoshuo
仿照了coco数据集的格式,修改了num_classes=2,命令如下: python -u tools/train.py -c configs/dcn/yolov3_r50vd_dcn_db_iouaware_obj365_pretrained_coco.yml -o pretrain_weights=/home/ResNet50_vd_dcn_db_obj365_pretrained
报错信息如下:
Error Message Summary:
Error: The input's size along the split dimension must be evenly divisible by Attr(num_or_sections). But received Attr(num_or_sections) = 8, input(X)'s shape = [2, 50, 4], Attr(dim) = 0.