实例分割Yolact边缘端部署 （二） 训练自己的模型-＞ onnx

yolact训练

1. 文件结构

• 首先从github或者Gitee下载Yolact源码
  • Github
  • Gitee
• 配置文件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