detectron2注册数据集

首先在/detectron2/detectron2/engine/defaults/下配置config-file的路径

detectron2注册数据集_第1张图片

在目录/detectron2/configs/COCO-InstanceSegmentation/下配置

mask_rcnn_R_50_FPN_1x.yaml

detectron2注册数据集_第2张图片

发现其BASE为Base-RCNN-C4.yaml

于是在/detectron2/configs/下配置Base-RCNN-C4.yaml

detectron2注册数据集_第3张图片

配置其中的DATASETS,然后在builtin.py中

根据选择的数据集,设置相关的路径

detectron2注册数据集_第4张图片

最后在/detectron2/detectron2/data/datasets/builtin.py中修改dataset的root目录

detectron2注册数据集_第5张图片

 

 

你可能感兴趣的:(detectron2,b站)