search

Viredery / tf-eager-fasterrcnn

Faster R-CNN R-101-FPN model was implemented with TensorFlow2.0 eager execution.
MIT License
89 stars 46 forks source link

问题请教 #16

Closed cl886699 closed 4 years ago

cl886699 commented 4 years ago

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