facebookresearch / Detectron

FAIR's research platform for object detection research, implementing popular algorithms like Mask R-CNN and RetinaNet.
Apache License 2.0
26.27k stars 5.46k forks source link

ResNeXt 152 #848

Closed dorarad closed 5 years ago

dorarad commented 5 years ago

How can I run the model using the ResNeXt features? I tried to run the model with FPN.CONV_BODY: FPN.add_fpn_ResNet152_conv5_body and with the pretrained X-152-32x8d-IN5k.pkl file and got an assertion error: "Workspace blob res2_0_branch2a_w with shape (64, 64, 1, 1) does not match weights file shape (256, 64, 1, 1)"

So I believe there should be another conv body for the ResNeXt case? couldn't find it in FPN so I'm not sure what value I should set it to.

Thanks a lot!

dorarad commented 5 years ago