实例分割Yolact边缘端部署 (二) 训练自己的模型-> onnx
yolact训练
1. 文件结构
- 首先从github或者Gitee下载Yolact源码
- 配置文件
data/config.py
用来配置模型的所有参数,包括类别、训练数据、网络结构、超参数等等 eval.py
用来进行推理,包括后面模型转换也要用到train.py
训练模型
主要文件结构
.
├── data
│ ├── config.py
│ ├── __init__.py
│ ├── json2coco.py
│ ├── train_datasets
│ ├── yolact_example_0.png
│ ├── yolact_example_1.png
│ └── yolact_example_2.png
├── model
│ └── yolact_resnet50_54_800000.pth
├── backbone.py
├── eval.py
├── train.py
└── yolact.py
2. 修改config.py
2.1 与数据相关参数
- 修改类别,数据集中有两类,所以对应修改即可
COCO_CLASSES = ('cat','dog&#