search

jwyang / faster-rcnn.pytorch

A faster pytorch implementation of faster r-cnn
MIT License
7.66k stars 2.33k forks source link

Request for cfgs/res152.yml #588

Open keineahnung2345 opened 5 years ago

keineahnung2345 commented 5 years ago

From trainval_net.py, it seems res152 is supported, but when I run it with --net res152, it says the file cfgs/res152.yml cannot be found. So where is res152.yml?

AlexanderHustinx commented 5 years ago

That file doens't exist, but you can just make it yourself. Just have a look at the settings in res101.yml for inspiration.

By the way, you will need to probably change a few more things in the code for this to work correctly. e.g. in resnet.py's class resnet(_fasterRCNN) in function init_modules(self) you'll need to change: resnet = resnet101() into resnet = resnet152()

YangHangWu commented 4 years ago

That file doens't exist, but you can just make it yourself. Just have a look at the settings in res101.yml for inspiration.

By the way, you will need to probably change a few more things in the code for this to work correctly. e.g. in resnet.py's class resnet(_fasterRCNN) in function init_modules(self) you'll need to change: resnet = resnet101() into resnet = resnet152()

could you provide your own resnet152 cfg for example

AlexanderHustinx commented 4 years ago

I don't use ResNet-152 But I imagine you can just copy the yml from resnet101 and it will likely work