Обнаружение объектов в detectron2 с помощью pytorch в Google Colab. Повторно использовать уже обученную модель или импортировать существующую обученную модель и прогнозировать объекты

  1. Сначала я загрузил папку вывода обученной модели и импортировал ее в новый проект на сервере google colab.
  2. В новом проекте без обучения модели я указал путь 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". но все же он не предсказывает объекты.
  3. Я изменил путь весов модели и дал ранее обученные метрики модели 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

0 ответов

Другие вопросы по тегам