地址链接:https://www.flyai.com/m/efficientdet-d2.pth
查看来源:https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch
确定自己使用的框架并导入对应的库。导入库实现样例代码可参考 文档中心-预训练模型使用教程
在代码中实现加载预训练模型地址
# 必须使用该方法下载模型,然后加载
from flyai.utils import remote_helper
path = remote_helper.get_remote_data('https://www.flyai.com/m/efficientdet-d2.pth')
model.load_state_dict(torch.load(f'efficientdet-d2.pth'))
#模型定义请看考 https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch
模型加载请参考:https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch efficientdet_test.py