问题请教

[cl886699]

你好，我想用自己的数据训练模型，我数据的类别只有7种，我想要用你的weights，我想到的是修改bbox_head.py，但是发现还是得修改gaster_rcnn.py文件，感觉修改的地方太多了，有点改不过来。想知道有没有其他方法，比如加载weights的时候，不加载最后的分类和回归层。期待你的解答，谢谢