maskrcnn-benchmark-master（一）：配置文件

最新推荐文章于 2025-10-02 15:31:51 发布

原创

最新推荐文章于 2025-10-02 15:31:51 发布 · 1.4k 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#python #计算机视觉 #人工智能 #深度学习 #目标检测

前言

作为本系列的开篇，我首先介绍的是配置文件部分，因为整个模型的搭建都和配置文件有很大的关系，所以我们先来了解一下配置文件里面都有啥相关参数配置。

一、配置文件

maskrcnn-benckmark-master是提供有默认的配置的，见文件:your_project/maskrcnn_benckmark/config/defaults.py

import os
# 通过yacs包来生成的默认配置
from yacs.config import CfgNode as CN


_C = CN()

_C.MODEL = CN()
# 表示是否只有RPN
_C.MODEL.RPN_ONLY = False
# 表示是否含有Mask分支
_C.MODEL.MASK_ON = False
_C.MODEL.RETINANET_ON = False
_C.MODEL.KEYPOINT_ON = False
# 表示是否使用GPU
_C.MODEL.DEVICE = "cuda"
# 默认的网络框架
_C.MODEL.META_ARCHITECTURE = "GeneralizedRCNN"
_C.MODEL.CLS_AGNOSTIC_BBOX_REG = False


...
...

那如果我们需要修改相关配置，难道需要在这个defaults.py文件中进行修改吗？

NO! NO! NO!

我们其实只要把需要修改的配置写在一个带.yaml后缀的文件中，从而通过这个文件来修改默认的配置，见文件your_project/configs/e2e_mask_rcnn_R_50