Spaces:
Running
on
T4
Running
on
T4
data_aug_scales = [480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800] | |
data_aug_max_size = 1333 | |
data_aug_scales2_resize = [400, 500, 600] | |
data_aug_scales2_crop = [384, 600] | |
data_aug_scale_overlap = None | |
batch_size = 4 | |
modelname = 'groundingdino' | |
backbone = 'swin_B_384_22k' | |
position_embedding = 'sine' | |
pe_temperatureH = 20 | |
pe_temperatureW = 20 | |
return_interm_indices = [1, 2, 3] | |
enc_layers = 6 | |
dec_layers = 6 | |
pre_norm = False | |
dim_feedforward = 2048 | |
hidden_dim = 256 | |
dropout = 0.0 | |
nheads = 8 | |
num_queries = 900 | |
query_dim = 4 | |
num_patterns = 0 | |
num_feature_levels = 4 | |
enc_n_points = 4 | |
dec_n_points = 4 | |
two_stage_type = 'standard' | |
two_stage_bbox_embed_share = False | |
two_stage_class_embed_share = False | |
transformer_activation = 'relu' | |
dec_pred_bbox_embed_share = True | |
dn_box_noise_scale = 1.0 | |
dn_label_noise_ratio = 0.5 | |
dn_label_coef = 1.0 | |
dn_bbox_coef = 1.0 | |
embed_init_tgt = True | |
dn_labelbook_size = 91 | |
max_text_len = 256 | |
text_encoder_type = 'checkpoints/bert-base-uncased' | |
use_text_enhancer = True | |
use_fusion_layer = True | |
use_checkpoint = True | |
use_transformer_ckpt = True | |
use_text_cross_attention = True | |
text_dropout = 0.0 | |
fusion_dropout = 0.0 | |
fusion_droppath = 0.1 | |
sub_sentence_present = True | |
max_labels = 90 | |
lr = 0.0001 | |
backbone_freeze_keywords = None | |
freeze_keywords = ['backbone.0', 'bert'] | |
lr_backbone = 1e-05 | |
lr_backbone_names = ['backbone.0', 'bert'] | |
lr_linear_proj_mult = 1e-05 | |
lr_linear_proj_names = ['ref_point_head', 'sampling_offsets'] | |
weight_decay = 0.0001 | |
param_dict_type = 'ddetr_in_mmdet' | |
ddetr_lr_param = False | |
epochs = 30 | |
lr_drop = 10 | |
save_checkpoint_interval = 10 | |
clip_max_norm = 0.1 | |
onecyclelr = False | |
multi_step_lr = False | |
lr_drop_list = [10, 20] | |
frozen_weights = None | |
dilation = False | |
pdetr3_bbox_embed_diff_each_layer = False | |
pdetr3_refHW = -1 | |
random_refpoints_xy = False | |
fix_refpoints_hw = -1 | |
dabdetr_yolo_like_anchor_update = False | |
dabdetr_deformable_encoder = False | |
dabdetr_deformable_decoder = False | |
use_deformable_box_attn = False | |
box_attn_type = 'roi_align' | |
dec_layer_number = None | |
decoder_layer_noise = False | |
dln_xy_noise = 0.2 | |
dln_hw_noise = 0.2 | |
add_channel_attention = False | |
add_pos_value = False | |
two_stage_pat_embed = 0 | |
two_stage_add_query_num = 0 | |
two_stage_learn_wh = False | |
two_stage_default_hw = 0.05 | |
two_stage_keep_all_tokens = False | |
num_select = 900 | |
batch_norm_type = 'FrozenBatchNorm2d' | |
masks = False | |
aux_loss = True | |
set_cost_class = 5.0 | |
set_cost_bbox = 1.0 | |
set_cost_giou = 0.0 | |
cls_loss_coef = 5.0 | |
bbox_loss_coef = 1.0 | |
giou_loss_coef = 0.0 | |
enc_loss_coef = 1.0 | |
interm_loss_coef = 1.0 | |
no_interm_box_loss = False | |
mask_loss_coef = 1.0 | |
dice_loss_coef = 1.0 | |
focal_alpha = 0.25 | |
focal_gamma = 2.0 | |
decoder_sa_type = 'sa' | |
matcher_type = 'HungarianMatcher' | |
decoder_module_seq = ['sa', 'ca', 'ffn'] | |
nms_iou_threshold = -1 | |
dec_pred_class_embed_share = True | |
match_unstable_error = True | |
use_detached_boxes_dec_out = False | |
dn_scalar = 100 | |
box_threshold = 0.23 | |
text_threshold = 0 | |
use_coco_eval = False | |
label_list = [ | |
'alcohol bottle', 'baguette roll', 'ball', 'banana', 'bead', 'bee', | |
'birthday candle', 'biscuit', 'boat', 'bottle', 'bowl', 'box', | |
'bread roll', 'brick', 'buffalo', 'bun', 'calamari ring', 'can', 'candle', | |
'cap', 'car', 'cartridge', 'cassette', 'cement bag', 'cereal', | |
'chewing gum piece', 'chopstick', 'clam', 'coffee bean', 'coin', | |
'cotton ball', 'cow', 'crane', 'crayon', 'croissant', 'crow', 'cup', | |
'cupcake', 'cupcake holder', 'fish', 'gemstone', 'go game piece', 'goat', | |
'goldfish snack', 'goose', 'ice cream', 'ice cream cone', 'instant noodle', | |
'jade stone', 'jeans', 'kidney bean', 'kitchen towel', 'lighter', | |
'lipstick', 'm&m piece', 'macaron', 'match', 'meat skewer', 'mini blind', | |
'mosaic tile', 'naan bread', 'nail', 'nut', 'onion ring', 'orange', | |
'pearl', 'pen', 'pencil', 'penguin', 'pepper', 'person', 'pigeon', 'plate', | |
'polka dot tile', 'potato', 'rice bag', 'roof tile', 'screw', 'shoe', | |
'spoon', 'spring roll', 'stair', 'stapler pin', 'straw', | |
'supermarket shelf', 'swan', 'tomato', 'watermelon', 'window', 'zebra' | |
] | |
val_label_list = [ | |
'ant', 'bird', 'book', 'bottle cap', 'bullet', 'camel', 'chair', | |
'chicken wing', 'donut', 'donut holder', 'flamingo', 'flower', | |
'flower pot', 'grape', 'horse', 'kiwi', 'milk carton', 'oyster', | |
'oyster shell', 'package of fresh cut fruit', 'peach', 'pill', 'polka dot', | |
'prawn cracker', 'sausage', 'seagull', 'shallot', 'shirt', 'skateboard', | |
'toilet paper roll' | |
] | |