Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

config file #3

Open
mahilaMoghadami opened this issue Apr 5, 2023 · 1 comment
Open

config file #3

mahilaMoghadami opened this issue Apr 5, 2023 · 1 comment

Comments

@mahilaMoghadami
Copy link

mahilaMoghadami commented Apr 5, 2023

hello.
is there limitation in model selection for training?
As I check in mmdet/models/detector/obb/ models with OAN are Faster_rcnn, orientedRCNN, roi_transformer and retinanet.
so should we select one of above for custome confile? and we cant use all models available in mmdetection?

this part I mean:
model = dict(
type='oan_OrientedRCNN',
pretrained='torchvision://resnet50',
backbone=dict(
type='ResNet',
depth=50,
num_stages=4,
out_indices=(0, 1, 2, 3),
frozen_stages=1,
norm_cfg=dict(type='BN', requires_grad=True),
norm_eval=True,
style='pytorch'),

oan = dict(

    type= 'standard_OAN',
    oan_use_layer = 3,
    grid_num = 16, 
    oan_fc_hidden = 512,
    act_func = 'GELU',
    loss_oan_cls=dict(
        type='FocalLoss',
        use_sigmoid=True,
        gamma=2.0,
        alpha=0.25,
        loss_weight=8.0),
    filter_thr = 0.18    
@mahilaMoghadami mahilaMoghadami changed the title train new model config file Apr 5, 2023
@zhangpeng2001
Copy link
Collaborator

@mahilaMoghadami different models have their own attributes, you can read the realated papers and choose the best for your task. OAN could be added to mojarity plenty of existing models, like Oriented RCNN, Faster RCNN, RoI Trandformer and RetinaNet that we have uploaded, or you can add OAN to your custom models via referring to mmlab official documentation to change your models (https://mmdetection.readthedocs.io/en/latest/advanced_guides/customize_models.html).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants