基於detectron2的centermask2的加載權重檢測問題、數據集檢測classname等模型配置信息的加載,訓練過程以及檢測文件說明
一般流程
- 根據配置文件生成模型，使用訓練權重的模型配置模型、否則會出現模型和權重不符問題：例如權重是3個類別的，模型是80個類別就會導致此類問題。
- 加載已經訓練好的權重
- 進行檢測
模型配置信息加載，calssname等信息
通過train或者test注冊的數據集信息加載這些信息進框架。
最后效果文件以及流程文件
最后效果文件以及流程文件
訓練使用文件：testtraindemo下的custom_train_template.py進行train和eval評估；
檢測以及掩膜圖整理文件：demo文件夾下的demo.py