search

Ranchosky / OAN

Apache License 2.0
56 stars 7 forks source link

config file #3

Open mahilaMoghadami opened 1 year ago

mahilaMoghadami commented 1 year ago

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
zhangpeng2001 commented 1 year ago

@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).