MMDet
MMDet 安装
前置条件: nvcc、gcc的安装
安装 mmdet
pip install -U openmim
mim install mmengine
mim install mmcv
git clone https://github.com/open-mmlab/mmdetection.git
mim install -e mmdetection
配置文件(RTMDet)
metainfo
用于数据加载器中,例:
metainfo = {
# 类别信息
"classes": ("cat", "dog", ...),
# 可视化时候的RGB颜色
"palette": [
(255, 0, 0),
(0, 255, 0),
...
]
}
train_dataloader = dict(
dataset