Take a look at pytorch's official FasterRCNN code

Original text: https://zhuanlan.zhihu.com/p/145842317
Reference: http://t.csdn.cn/idjzU

View version and source code location in pycharm

## I found that there is no need to look carefully at the functions in this version, and focus on how to implement them

import torchvision

print(torchvision.__version__)
# 0.12.0
print(torchvision.__path__)
# ['D:\\anaconda\\lib\\site-packages\\torchvision']

code structure

As the base class of target detection in torchvision, GeneralizedRCNN