保存经过训练的 Detectron2 模型并对单个图像进行预测

问题描述

我是detectron2 的新手，这是我的第一个项目。在阅读文档并使用教程作为指导后，我在自定义数据集上训练了我的模型并进行了评估。

我现在想通过加载这个保存的模型对我通过 API 接收的图像进行预测。我找不到任何可以帮助我完成这项任务的阅读材料。

为了保存我的模型，我使用此链接作为参考 - https://detectron2.readthedocs.io/en/latest/tutorials/models.html

我可以使用以下代码保存我的训练模型-

from detectron2.modeling import build_model
model = build_model(cfg) # returns a torch.nn.Module

from detectron2.checkpoint import DetectionCheckpointer

checkpointer = DetectionCheckpointer(model,save_dir="output")
checkpointer.save("model_final") # save to output/model_final.pth

但我仍然对如何实现我想要的东西感到困惑。我可以使用一些关于下一步应该做什么的指导。非常感谢任何可以提供帮助的人。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）