pytorch手册中 微调基于torchvision 0.3的目标检测模型一章:
4.整合
在 references/detection/ 中,我们有许多辅助函数来简化训练和评估检测模型。在这里,我们 将使用 references/detection/engine.py , references/detection/utils.py 和 references/ detection/transforms.py 。 只需将它们复制到您的文件夹并在此处使用它们
其中references/detection/可以在
https://github.com/pytorch/vision/tree/main/references/detection中下载得到
或references.zip-深度学习文档类资源-CSDN下载 下载得到
除此之外,还可能产生’No module named 'pycocotools' 错误或者No module named 'pycocotools._mask'等错误,使用下语句安装pycocotools,重启内核即可!
!pip install pycocotools