Translating Detectron1 config file to Detectron2? #3478
Unanswered
cooldecola
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Hey guys, I have a yaml file in detectron that I used to train a model. I am now trying to convert to detectron2 and can't seem to find the same parameters in detectron2.
This is the yaml file in detectron:
D1 = detectron1
D2 = detectron2
Problem 1
In D1 I set
CONV_BODY: FPN.add_fpn_ResNet50_conv5_bodyhowever in D2 I have it set ascfg.merge_from_file(model_zoo.get_config_file("Base-RCNN-FPN.yaml")). Inside Base-RCNN-FPN.yaml I seeAre these backbones the same?
build_resnet_fpn_backbonedoes not specify if it is ResNet50Problem 2
How do I set how many GPU's ? In D1 I have
NUM_GPUS: 1but nothing similar in D2?Problem 3
Under the FPN section of my yaml config file I have set
I don't have the option to use FPN for RoI transforms or using FPN for RPN. Is there a way to do this?
Problem 4
Same issue with the
FAST_RCNNoptions. I cannot find similar options that translate over.The following is what I have implemented so far in D2 but I'm not sure if this is correct:
Last Problem
I have set
BATCH_SIZE_PER_IM = 256in D1.Is this the same as c
fg.MODEL.BATCH_SIZE_PER_IMAGE = 256?The reason I am asking is that in D1 the BATCH_SIZE_PER_IM falls under the training category, however in D2 it's under the model category. I think they should be the same but I just wanted to clarify.
Any help would be greatly appreciated, thanks in advance! :)
Beta Was this translation helpful? Give feedback.
All reactions