Обнаружение объектов в detectron2 с помощью pytorch в Google Colab. Повторно использовать уже обученную модель или импортировать существующую обученную модель и прогнозировать объекты
- Сначала я загрузил папку вывода обученной модели и импортировал ее в новый проект на сервере google colab.
- В новом проекте без обучения модели я указал путь
model_final.pth
существующей выходной папки вcfg.MODEL.WEIGHTS
=/ содержание / вывод /model_final.pth. но идет в бесконечный цикл. 3. меняю веса моделиcfg.MODEL.WEIGHTS = "detectron2://COCO-Detection/faster_rcnn_R_101_FPN_3x/137851257/model_final_f6e8b1.pkl"
. но все же он не предсказывает объекты. - Я изменил путь весов модели и дал ранее обученные метрики модели
JSON
файл все еще не работаетcfg.MODEL.WEIGHTS=/content/output/metrics.json 5.By using DetectionCheckpointer(model).load("/content/output/model_final.pth") DetectionCheckpointer(model).load("detectron2://COCO-Detection/faster_rcnn_R_101_FPN_3x/137851257/model_final_f6e8b1.pkl")
это дает ошибку, модель не определена. что этоmodel_final.pkl
файл? а где мы это взяли? что нам делать, чтобы импортировать существующую модель поезда и спрогнозировать объекты в новом проекте?
cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR, "model_final.pth")
cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
cfg.DATASETS.TEST = ("microcontroller_test", )
predictor = DefaultPredictor(cfg)
Вышеупомянутый код переходит в бесконечный цикл
DetectionCheckpointer(model).load("/content/output/model_final.pth")
DetectionCheckpointer(model).load("detectron2://COCO-Detection/faster_rcnn_R_101_FPN_3x/137851257/model_final_f6e8b1.pkl")
Ошибка:
NameError Traceback (most recent call last)
<ipython-input-12-69f2a7846756> in <module>()
----> 1 DetectionCheckpointer(model).load("/content/output/model_final.pth")
2
3 DetectionCheckpointer(model).load("detectron2://COCO-Detection/faster_rcnn_R_101_FPN_3x/137851257/model_final_f6e8b1.pkl")
NameError: name 'model' is not defined