此处先说明一下,我更换的是“ResNet1”,之前用的是“ResNet”。
这个是需要修改的地方,记得只需要修改“init.py”和“backbone_new.py”这两个文件,
我是直接用的原来的“resnet.py”代码,然后自己重新新建了一个“backbone_new.py”文件,调用就可以。当然,记得在自己调用的类上面添加
@BACKBONES.register_module()
这个部分,显示如下:
因为,我为了区别是调用的“ResNet”还是自己的“ResNet1”,我把所有“backbone_new.py”文件中的“ResNet”改称了“ResNet1”,
记得修改自己的网络结构代码。调用自己的“ResNet1”的backbone结构,内容如下所示:
同时,记得重新编译自己的代码:
python setup.py develop
貌似没其他的地方需要修改的了。