mask 训练自己数据集出错。包括train.py 和infer.py
收藏
使用命令启动 python tools/infer.py -c configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.yml --infer_img=demo/6.jpg --output_dir=infer_output/ --draw_threshold=0.5 -o weights=output/mask_rcnn_r50_fpn_1x_coco/model_final --use_vdl=True
0
收藏
请登录后评论
还有更多信息没
从报错上来看，应该是你数据集的问题，应该是有的图片没读进来，检测了一个空值，就报错了。