diff --git "a/configs/config_cascade_rcnn_traffic_signs.py" "b/configs/config_cascade_rcnn_traffic_signs.py" new file mode 100644--- /dev/null +++ "b/configs/config_cascade_rcnn_traffic_signs.py" @@ -0,0 +1,6157 @@ +model = dict( + type='CascadeRCNN', + data_preprocessor=dict( + type='DetDataPreprocessor', + mean=[ + 123.675, + 116.28, + 103.53, + ], + std=[ + 58.395, + 57.12, + 57.375, + ], + bgr_to_rgb=True, + pad_size_divisor=32), + 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', + init_cfg=dict(type='Pretrained', checkpoint='torchvision://resnet50')), + neck=dict( + type='FPN', + in_channels=[ + 256, + 512, + 1024, + 2048, + ], + out_channels=256, + num_outs=5), + rpn_head=dict( + type='RPNHead', + in_channels=256, + feat_channels=256, + anchor_generator=dict( + type='AnchorGenerator', + scales=[ + 8, + ], + ratios=[ + 0.5, + 1.0, + 2.0, + ], + strides=[ + 4, + 8, + 16, + 32, + 64, + ]), + bbox_coder=dict( + type='DeltaXYWHBBoxCoder', + target_means=[ + 0.0, + 0.0, + 0.0, + 0.0, + ], + target_stds=[ + 1.0, + 1.0, + 1.0, + 1.0, + ]), + loss_cls=dict( + type='CrossEntropyLoss', use_sigmoid=True, loss_weight=1.0), + loss_bbox=dict( + type='SmoothL1Loss', beta=0.1111111111111111, loss_weight=1.0)), + roi_head=dict( + type='CascadeRoIHead', + num_stages=3, + stage_loss_weights=[ + 1, + 0.5, + 0.25, + ], + bbox_roi_extractor=dict( + type='SingleRoIExtractor', + roi_layer=dict(type='RoIAlign', output_size=7, sampling_ratio=0), + out_channels=256, + featmap_strides=[ + 4, + 8, + 16, + 32, + ]), + bbox_head=[ + dict( + type='Shared2FCBBoxHead', + in_channels=256, + fc_out_channels=1024, + roi_feat_size=7, + num_classes=237, + bbox_coder=dict( + type='DeltaXYWHBBoxCoder', + target_means=[ + 0.0, + 0.0, + 0.0, + 0.0, + ], + target_stds=[ + 0.1, + 0.1, + 0.2, + 0.2, + ]), + reg_class_agnostic=True, + loss_cls=dict( + type='CrossEntropyLoss', + use_sigmoid=False, + loss_weight=1.0), + loss_bbox=dict(type='SmoothL1Loss', beta=1.0, + loss_weight=1.0)), + dict( + type='Shared2FCBBoxHead', + in_channels=256, + fc_out_channels=1024, + roi_feat_size=7, + num_classes=237, + bbox_coder=dict( + type='DeltaXYWHBBoxCoder', + target_means=[ + 0.0, + 0.0, + 0.0, + 0.0, + ], + target_stds=[ + 0.05, + 0.05, + 0.1, + 0.1, + ]), + reg_class_agnostic=True, + loss_cls=dict( + type='CrossEntropyLoss', + use_sigmoid=False, + loss_weight=1.0), + loss_bbox=dict(type='SmoothL1Loss', beta=1.0, + loss_weight=1.0)), + dict( + type='Shared2FCBBoxHead', + in_channels=256, + fc_out_channels=1024, + roi_feat_size=7, + num_classes=237, + bbox_coder=dict( + type='DeltaXYWHBBoxCoder', + target_means=[ + 0.0, + 0.0, + 0.0, + 0.0, + ], + target_stds=[ + 0.033, + 0.033, + 0.067, + 0.067, + ]), + reg_class_agnostic=True, + loss_cls=dict( + type='CrossEntropyLoss', + use_sigmoid=False, + loss_weight=1.0), + loss_bbox=dict(type='SmoothL1Loss', beta=1.0, + loss_weight=1.0)), + ]), + train_cfg=dict( + rpn=dict( + assigner=dict( + type='MaxIoUAssigner', + pos_iou_thr=0.7, + neg_iou_thr=0.3, + min_pos_iou=0.3, + match_low_quality=True, + ignore_iof_thr=-1), + sampler=dict( + type='RandomSampler', + num=256, + pos_fraction=0.5, + neg_pos_ub=-1, + add_gt_as_proposals=False), + allowed_border=0, + pos_weight=-1, + debug=False), + rpn_proposal=dict( + nms_pre=2000, + max_per_img=2000, + nms=dict(type='nms', iou_threshold=0.7), + min_bbox_size=0), + rcnn=[ + dict( + assigner=dict( + type='MaxIoUAssigner', + pos_iou_thr=0.5, + neg_iou_thr=0.5, + min_pos_iou=0.5, + match_low_quality=False, + ignore_iof_thr=-1), + sampler=dict( + type='RandomSampler', + num=512, + pos_fraction=0.25, + neg_pos_ub=-1, + add_gt_as_proposals=True), + pos_weight=-1, + debug=False), + dict( + assigner=dict( + type='MaxIoUAssigner', + pos_iou_thr=0.6, + neg_iou_thr=0.6, + min_pos_iou=0.6, + match_low_quality=False, + ignore_iof_thr=-1), + sampler=dict( + type='RandomSampler', + num=512, + pos_fraction=0.25, + neg_pos_ub=-1, + add_gt_as_proposals=True), + pos_weight=-1, + debug=False), + dict( + assigner=dict( + type='MaxIoUAssigner', + pos_iou_thr=0.7, + neg_iou_thr=0.7, + min_pos_iou=0.7, + match_low_quality=False, + ignore_iof_thr=-1), + sampler=dict( + type='RandomSampler', + num=512, + pos_fraction=0.25, + neg_pos_ub=-1, + add_gt_as_proposals=True), + pos_weight=-1, + debug=False), + ]), + test_cfg=dict( + rpn=dict( + nms_pre=1000, + max_per_img=1000, + nms=dict(type='nms', iou_threshold=0.7), + min_bbox_size=0), + rcnn=dict( + score_thr=0.05, + nms=dict(type='nms', iou_threshold=0.5), + max_per_img=100))) +dataset_type = 'CocoDataset' +data_root = '/storage/plzen4-ntis/projects/korpusy_cv/Didymos/Traffic_signs_new' +backend_args = None +train_pipeline = [ + dict(type='LoadImageFromFile', backend_args=None), + dict(type='LoadAnnotations', with_bbox=True), + dict(type='Resize', scale=( + 1333, + 800, + ), keep_ratio=True), + dict(type='RandomFlip', prob=0.5), + dict(type='PackDetInputs'), +] +test_pipeline = [ + dict(type='LoadImageFromFile', backend_args=None), + dict(type='Resize', scale=( + 1333, + 800, + ), keep_ratio=True), + dict(type='LoadAnnotations', with_bbox=True), + dict( + type='PackDetInputs', + meta_keys=( + 'img_id', + 'img_path', + 'ori_shape', + 'img_shape', + 'scale_factor', + )), +] +train_dataloader = dict( + batch_size=4, + num_workers=8, + persistent_workers=True, + sampler=dict(type='DefaultSampler', shuffle=True), + batch_sampler=dict(type='AspectRatioBatchSampler'), + dataset=dict( + type='CocoDataset', + data_root= + '/storage/plzen4-ntis/projects/korpusy_cv/Didymos/Traffic_signs_new', + ann_file='metadata_train_enriched.json', + data_prefix=dict(img='images/'), + filter_cfg=dict(filter_empty_gt=True, min_size=32), + pipeline=[ + dict(type='LoadImageFromFile', backend_args=None), + dict(type='LoadAnnotations', with_bbox=True), + dict(type='Resize', scale=( + 1333, + 800, + ), keep_ratio=True), + dict(type='RandomFlip', prob=0.5), + dict(type='PackDetInputs'), + ], + backend_args=None, + metainfo=dict( + classes=( + 'A10', + 'A11', + 'A12', + 'A14', + 'A15', + 'A16', + 'A17', + 'A18', + 'A19', + 'A1a', + 'A1b', + 'A22', + 'A24', + 'A28', + 'A29', + 'A2a', + 'A2b', + 'A30', + 'A31a', + 'A31b', + 'A31c', + 'A32a', + 'A32b', + 'A4', + 'A5a', + 'A6a', + 'A6b', + 'A7a', + 'A8', + 'A9', + 'B1', + 'B11', + 'B12', + 'B13', + 'B14', + 'B15', + 'B16', + 'B17', + 'B19', + 'B2', + 'B20a', + 'B20b', + 'B21a', + 'B21b', + 'B24a', + 'B24b', + 'B26', + 'B28', + 'B29', + 'B32', + 'B4', + 'B5', + 'B6', + 'C1', + 'C10a', + 'C10b', + 'C13a', + 'C14a', + 'C2a', + 'C2b', + 'C2c', + 'C2d', + 'C2e', + 'C2f', + 'C3a', + 'C3b', + 'C4a', + 'C4b', + 'C4c', + 'C7a', + 'C9a', + 'C9b', + 'E1', + 'E11', + 'E11c', + 'E12', + 'E13', + 'E2a', + 'E2b', + 'E2c', + 'E2d', + 'E3a', + 'E3b', + 'E4', + 'E5', + 'E6', + 'E7a', + 'E7b', + 'E8a', + 'E8b', + 'E8c', + 'E8d', + 'E8e', + 'E9', + 'I2', + 'IJ1', + 'IJ10', + 'IJ11a', + 'IJ11b', + 'IJ14c', + 'IJ15', + 'IJ2', + 'IJ3', + 'IJ4a', + 'IJ4b', + 'IJ4c', + 'IJ4d', + 'IJ4e', + 'IJ5', + 'IJ6', + 'IJ7', + 'IJ8', + 'IJ9', + 'IP10a', + 'IP10b', + 'IP11a', + 'IP11b', + 'IP11c', + 'IP11e', + 'IP11g', + 'IP12', + 'IP13a', + 'IP13b', + 'IP13c', + 'IP13d', + 'IP14a', + 'IP15a', + 'IP15b', + 'IP16', + 'IP17', + 'IP18a', + 'IP18b', + 'IP19', + 'IP2', + 'IP21', + 'IP21a', + 'IP22', + 'IP25a', + 'IP25b', + 'IP26a', + 'IP26b', + 'IP27a', + 'IP3', + 'IP31a', + 'IP4a', + 'IP4b', + 'IP5', + 'IP6', + 'IP7', + 'IP8a', + 'IP8b', + 'IS10b', + 'IS11a', + 'IS11b', + 'IS11c', + 'IS12a', + 'IS12b', + 'IS12c', + 'IS13', + 'IS14', + 'IS15a', + 'IS15b', + 'IS16b', + 'IS16c', + 'IS16d', + 'IS17', + 'IS18a', + 'IS18b', + 'IS19a', + 'IS19b', + 'IS19c', + 'IS19d', + 'IS1a', + 'IS1b', + 'IS1c', + 'IS1d', + 'IS20', + 'IS21a', + 'IS21b', + 'IS21c', + 'IS22a', + 'IS22c', + 'IS22d', + 'IS22e', + 'IS22f', + 'IS23', + 'IS24a', + 'IS24b', + 'IS24c', + 'IS2a', + 'IS2b', + 'IS2c', + 'IS2d', + 'IS3a', + 'IS3b', + 'IS3c', + 'IS3d', + 'IS4a', + 'IS4b', + 'IS4c', + 'IS4d', + 'IS5', + 'IS6a', + 'IS6b', + 'IS6c', + 'IS6e', + 'IS6f', + 'IS6g', + 'IS7a', + 'IS8a', + 'IS8b', + 'IS9a', + 'IS9b', + 'IS9c', + 'IS9d', + 'O2', + 'P1', + 'P2', + 'P3', + 'P4', + 'P6', + 'P7', + 'P8', + 'UNKNOWN', + 'X1', + 'X2', + 'X3', + 'XXX', + 'Z2', + 'Z3', + 'Z4a', + 'Z4b', + 'Z4c', + 'Z4d', + 'Z4e', + 'Z7', + 'Z9', + ), + palette=[ + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ]))) +val_dataloader = dict( + batch_size=1, + num_workers=2, + persistent_workers=True, + drop_last=False, + sampler=dict(type='DefaultSampler', shuffle=False), + dataset=dict( + type='CocoDataset', + data_root= + '/storage/plzen4-ntis/projects/korpusy_cv/Didymos/Traffic_signs_new', + ann_file='metadata_val_enriched.json', + data_prefix=dict(img='images/'), + test_mode=True, + pipeline=[ + dict(type='LoadImageFromFile', backend_args=None), + dict(type='Resize', scale=( + 1333, + 800, + ), keep_ratio=True), + dict(type='LoadAnnotations', with_bbox=True), + dict( + type='PackDetInputs', + meta_keys=( + 'img_id', + 'img_path', + 'ori_shape', + 'img_shape', + 'scale_factor', + )), + ], + backend_args=None, + metainfo=dict( + classes=( + 'A10', + 'A11', + 'A12', + 'A14', + 'A15', + 'A16', + 'A17', + 'A18', + 'A19', + 'A1a', + 'A1b', + 'A22', + 'A24', + 'A28', + 'A29', + 'A2a', + 'A2b', + 'A30', + 'A31a', + 'A31b', + 'A31c', + 'A32a', + 'A32b', + 'A4', + 'A5a', + 'A6a', + 'A6b', + 'A7a', + 'A8', + 'A9', + 'B1', + 'B11', + 'B12', + 'B13', + 'B14', + 'B15', + 'B16', + 'B17', + 'B19', + 'B2', + 'B20a', + 'B20b', + 'B21a', + 'B21b', + 'B24a', + 'B24b', + 'B26', + 'B28', + 'B29', + 'B32', + 'B4', + 'B5', + 'B6', + 'C1', + 'C10a', + 'C10b', + 'C13a', + 'C14a', + 'C2a', + 'C2b', + 'C2c', + 'C2d', + 'C2e', + 'C2f', + 'C3a', + 'C3b', + 'C4a', + 'C4b', + 'C4c', + 'C7a', + 'C9a', + 'C9b', + 'E1', + 'E11', + 'E11c', + 'E12', + 'E13', + 'E2a', + 'E2b', + 'E2c', + 'E2d', + 'E3a', + 'E3b', + 'E4', + 'E5', + 'E6', + 'E7a', + 'E7b', + 'E8a', + 'E8b', + 'E8c', + 'E8d', + 'E8e', + 'E9', + 'I2', + 'IJ1', + 'IJ10', + 'IJ11a', + 'IJ11b', + 'IJ14c', + 'IJ15', + 'IJ2', + 'IJ3', + 'IJ4a', + 'IJ4b', + 'IJ4c', + 'IJ4d', + 'IJ4e', + 'IJ5', + 'IJ6', + 'IJ7', + 'IJ8', + 'IJ9', + 'IP10a', + 'IP10b', + 'IP11a', + 'IP11b', + 'IP11c', + 'IP11e', + 'IP11g', + 'IP12', + 'IP13a', + 'IP13b', + 'IP13c', + 'IP13d', + 'IP14a', + 'IP15a', + 'IP15b', + 'IP16', + 'IP17', + 'IP18a', + 'IP18b', + 'IP19', + 'IP2', + 'IP21', + 'IP21a', + 'IP22', + 'IP25a', + 'IP25b', + 'IP26a', + 'IP26b', + 'IP27a', + 'IP3', + 'IP31a', + 'IP4a', + 'IP4b', + 'IP5', + 'IP6', + 'IP7', + 'IP8a', + 'IP8b', + 'IS10b', + 'IS11a', + 'IS11b', + 'IS11c', + 'IS12a', + 'IS12b', + 'IS12c', + 'IS13', + 'IS14', + 'IS15a', + 'IS15b', + 'IS16b', + 'IS16c', + 'IS16d', + 'IS17', + 'IS18a', + 'IS18b', + 'IS19a', + 'IS19b', + 'IS19c', + 'IS19d', + 'IS1a', + 'IS1b', + 'IS1c', + 'IS1d', + 'IS20', + 'IS21a', + 'IS21b', + 'IS21c', + 'IS22a', + 'IS22c', + 'IS22d', + 'IS22e', + 'IS22f', + 'IS23', + 'IS24a', + 'IS24b', + 'IS24c', + 'IS2a', + 'IS2b', + 'IS2c', + 'IS2d', + 'IS3a', + 'IS3b', + 'IS3c', + 'IS3d', + 'IS4a', + 'IS4b', + 'IS4c', + 'IS4d', + 'IS5', + 'IS6a', + 'IS6b', + 'IS6c', + 'IS6e', + 'IS6f', + 'IS6g', + 'IS7a', + 'IS8a', + 'IS8b', + 'IS9a', + 'IS9b', + 'IS9c', + 'IS9d', + 'O2', + 'P1', + 'P2', + 'P3', + 'P4', + 'P6', + 'P7', + 'P8', + 'UNKNOWN', + 'X1', + 'X2', + 'X3', + 'XXX', + 'Z2', + 'Z3', + 'Z4a', + 'Z4b', + 'Z4c', + 'Z4d', + 'Z4e', + 'Z7', + 'Z9', + ), + palette=[ + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ]))) +test_dataloader = dict( + batch_size=1, + num_workers=2, + persistent_workers=True, + drop_last=False, + sampler=dict(type='DefaultSampler', shuffle=False), + dataset=dict( + type='CocoDataset', + data_root= + '/storage/plzen4-ntis/projects/korpusy_cv/Didymos/Traffic_signs_new', + ann_file='metadata_val_enriched.json', + data_prefix=dict(img='images/'), + test_mode=True, + pipeline=[ + dict(type='LoadImageFromFile', backend_args=None), + dict(type='Resize', scale=( + 1333, + 800, + ), keep_ratio=True), + dict(type='LoadAnnotations', with_bbox=True), + dict( + type='PackDetInputs', + meta_keys=( + 'img_id', + 'img_path', + 'ori_shape', + 'img_shape', + 'scale_factor', + )), + ], + backend_args=None, + metainfo=dict( + classes=( + 'A10', + 'A11', + 'A12', + 'A14', + 'A15', + 'A16', + 'A17', + 'A18', + 'A19', + 'A1a', + 'A1b', + 'A22', + 'A24', + 'A28', + 'A29', + 'A2a', + 'A2b', + 'A30', + 'A31a', + 'A31b', + 'A31c', + 'A32a', + 'A32b', + 'A4', + 'A5a', + 'A6a', + 'A6b', + 'A7a', + 'A8', + 'A9', + 'B1', + 'B11', + 'B12', + 'B13', + 'B14', + 'B15', + 'B16', + 'B17', + 'B19', + 'B2', + 'B20a', + 'B20b', + 'B21a', + 'B21b', + 'B24a', + 'B24b', + 'B26', + 'B28', + 'B29', + 'B32', + 'B4', + 'B5', + 'B6', + 'C1', + 'C10a', + 'C10b', + 'C13a', + 'C14a', + 'C2a', + 'C2b', + 'C2c', + 'C2d', + 'C2e', + 'C2f', + 'C3a', + 'C3b', + 'C4a', + 'C4b', + 'C4c', + 'C7a', + 'C9a', + 'C9b', + 'E1', + 'E11', + 'E11c', + 'E12', + 'E13', + 'E2a', + 'E2b', + 'E2c', + 'E2d', + 'E3a', + 'E3b', + 'E4', + 'E5', + 'E6', + 'E7a', + 'E7b', + 'E8a', + 'E8b', + 'E8c', + 'E8d', + 'E8e', + 'E9', + 'I2', + 'IJ1', + 'IJ10', + 'IJ11a', + 'IJ11b', + 'IJ14c', + 'IJ15', + 'IJ2', + 'IJ3', + 'IJ4a', + 'IJ4b', + 'IJ4c', + 'IJ4d', + 'IJ4e', + 'IJ5', + 'IJ6', + 'IJ7', + 'IJ8', + 'IJ9', + 'IP10a', + 'IP10b', + 'IP11a', + 'IP11b', + 'IP11c', + 'IP11e', + 'IP11g', + 'IP12', + 'IP13a', + 'IP13b', + 'IP13c', + 'IP13d', + 'IP14a', + 'IP15a', + 'IP15b', + 'IP16', + 'IP17', + 'IP18a', + 'IP18b', + 'IP19', + 'IP2', + 'IP21', + 'IP21a', + 'IP22', + 'IP25a', + 'IP25b', + 'IP26a', + 'IP26b', + 'IP27a', + 'IP3', + 'IP31a', + 'IP4a', + 'IP4b', + 'IP5', + 'IP6', + 'IP7', + 'IP8a', + 'IP8b', + 'IS10b', + 'IS11a', + 'IS11b', + 'IS11c', + 'IS12a', + 'IS12b', + 'IS12c', + 'IS13', + 'IS14', + 'IS15a', + 'IS15b', + 'IS16b', + 'IS16c', + 'IS16d', + 'IS17', + 'IS18a', + 'IS18b', + 'IS19a', + 'IS19b', + 'IS19c', + 'IS19d', + 'IS1a', + 'IS1b', + 'IS1c', + 'IS1d', + 'IS20', + 'IS21a', + 'IS21b', + 'IS21c', + 'IS22a', + 'IS22c', + 'IS22d', + 'IS22e', + 'IS22f', + 'IS23', + 'IS24a', + 'IS24b', + 'IS24c', + 'IS2a', + 'IS2b', + 'IS2c', + 'IS2d', + 'IS3a', + 'IS3b', + 'IS3c', + 'IS3d', + 'IS4a', + 'IS4b', + 'IS4c', + 'IS4d', + 'IS5', + 'IS6a', + 'IS6b', + 'IS6c', + 'IS6e', + 'IS6f', + 'IS6g', + 'IS7a', + 'IS8a', + 'IS8b', + 'IS9a', + 'IS9b', + 'IS9c', + 'IS9d', + 'O2', + 'P1', + 'P2', + 'P3', + 'P4', + 'P6', + 'P7', + 'P8', + 'UNKNOWN', + 'X1', + 'X2', + 'X3', + 'XXX', + 'Z2', + 'Z3', + 'Z4a', + 'Z4b', + 'Z4c', + 'Z4d', + 'Z4e', + 'Z7', + 'Z9', + ), + palette=[ + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ]))) +val_evaluator = dict( + type='CocoMetric', + ann_file= + '/storage/plzen4-ntis/projects/korpusy_cv/Didymos/Traffic_signs_newmetadata_val_enriched.json', + metric='bbox', + format_only=False, + backend_args=None) +test_evaluator = dict( + type='CocoMetric', + ann_file= + '/storage/plzen4-ntis/projects/korpusy_cv/Didymos/Traffic_signs_newmetadata_val_enriched.json', + metric='bbox', + format_only=False, + backend_args=None) +train_cfg = dict(type='EpochBasedTrainLoop', max_epochs=20, val_interval=1) +val_cfg = dict(type='ValLoop') +test_cfg = dict(type='TestLoop') +param_scheduler = [ + dict( + type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500), + dict( + type='MultiStepLR', + begin=0, + end=20, + by_epoch=True, + milestones=[ + 16, + 19, + ], + gamma=0.1), +] +optim_wrapper = dict( + type='OptimWrapper', + optimizer=dict(type='SGD', lr=0.02, momentum=0.9, weight_decay=0.0001)) +auto_scale_lr = dict(enable=False, base_batch_size=16) +default_scope = 'mmdet' +default_hooks = dict( + timer=dict(type='IterTimerHook'), + logger=dict(type='LoggerHook', interval=50), + param_scheduler=dict(type='ParamSchedulerHook'), + checkpoint=dict(type='CheckpointHook', interval=1), + sampler_seed=dict(type='DistSamplerSeedHook'), + visualization=dict(type='DetVisualizationHook')) +env_cfg = dict( + cudnn_benchmark=False, + mp_cfg=dict(mp_start_method='fork', opencv_num_threads=0), + dist_cfg=dict(backend='nccl')) +vis_backends = [ + dict(type='LocalVisBackend'), + dict( + type='WandbVisBackend', + init_kwargs=dict(project='traffic_signs_cascadeRCNN')), +] +visualizer = dict( + type='DetLocalVisualizer', + vis_backends=[ + dict(type='LocalVisBackend'), + dict( + type='WandbVisBackend', + init_kwargs=dict(project='traffic_signs_cascadeRCNN')), + ], + name='visualizer') +log_processor = dict(type='LogProcessor', window_size=50, by_epoch=True) +log_level = 'INFO' +load_from = 'checkpoints/cascade_rcnn_traffic_signs_v1.pth' +resume = False +WANDB_PROJECT = 'traffic_signs_cascadeRCNN' +DATA_ROOT = '/storage/plzen4-ntis/projects/korpusy_cv/Didymos/Traffic_signs_new' +NUM_CLASSES = 237 +metainfo = dict( + classes=( + 'A10', + 'A11', + 'A12', + 'A14', + 'A15', + 'A16', + 'A17', + 'A18', + 'A19', + 'A1a', + 'A1b', + 'A22', + 'A24', + 'A28', + 'A29', + 'A2a', + 'A2b', + 'A30', + 'A31a', + 'A31b', + 'A31c', + 'A32a', + 'A32b', + 'A4', + 'A5a', + 'A6a', + 'A6b', + 'A7a', + 'A8', + 'A9', + 'B1', + 'B11', + 'B12', + 'B13', + 'B14', + 'B15', + 'B16', + 'B17', + 'B19', + 'B2', + 'B20a', + 'B20b', + 'B21a', + 'B21b', + 'B24a', + 'B24b', + 'B26', + 'B28', + 'B29', + 'B32', + 'B4', + 'B5', + 'B6', + 'C1', + 'C10a', + 'C10b', + 'C13a', + 'C14a', + 'C2a', + 'C2b', + 'C2c', + 'C2d', + 'C2e', + 'C2f', + 'C3a', + 'C3b', + 'C4a', + 'C4b', + 'C4c', + 'C7a', + 'C9a', + 'C9b', + 'E1', + 'E11', + 'E11c', + 'E12', + 'E13', + 'E2a', + 'E2b', + 'E2c', + 'E2d', + 'E3a', + 'E3b', + 'E4', + 'E5', + 'E6', + 'E7a', + 'E7b', + 'E8a', + 'E8b', + 'E8c', + 'E8d', + 'E8e', + 'E9', + 'I2', + 'IJ1', + 'IJ10', + 'IJ11a', + 'IJ11b', + 'IJ14c', + 'IJ15', + 'IJ2', + 'IJ3', + 'IJ4a', + 'IJ4b', + 'IJ4c', + 'IJ4d', + 'IJ4e', + 'IJ5', + 'IJ6', + 'IJ7', + 'IJ8', + 'IJ9', + 'IP10a', + 'IP10b', + 'IP11a', + 'IP11b', + 'IP11c', + 'IP11e', + 'IP11g', + 'IP12', + 'IP13a', + 'IP13b', + 'IP13c', + 'IP13d', + 'IP14a', + 'IP15a', + 'IP15b', + 'IP16', + 'IP17', + 'IP18a', + 'IP18b', + 'IP19', + 'IP2', + 'IP21', + 'IP21a', + 'IP22', + 'IP25a', + 'IP25b', + 'IP26a', + 'IP26b', + 'IP27a', + 'IP3', + 'IP31a', + 'IP4a', + 'IP4b', + 'IP5', + 'IP6', + 'IP7', + 'IP8a', + 'IP8b', + 'IS10b', + 'IS11a', + 'IS11b', + 'IS11c', + 'IS12a', + 'IS12b', + 'IS12c', + 'IS13', + 'IS14', + 'IS15a', + 'IS15b', + 'IS16b', + 'IS16c', + 'IS16d', + 'IS17', + 'IS18a', + 'IS18b', + 'IS19a', + 'IS19b', + 'IS19c', + 'IS19d', + 'IS1a', + 'IS1b', + 'IS1c', + 'IS1d', + 'IS20', + 'IS21a', + 'IS21b', + 'IS21c', + 'IS22a', + 'IS22c', + 'IS22d', + 'IS22e', + 'IS22f', + 'IS23', + 'IS24a', + 'IS24b', + 'IS24c', + 'IS2a', + 'IS2b', + 'IS2c', + 'IS2d', + 'IS3a', + 'IS3b', + 'IS3c', + 'IS3d', + 'IS4a', + 'IS4b', + 'IS4c', + 'IS4d', + 'IS5', + 'IS6a', + 'IS6b', + 'IS6c', + 'IS6e', + 'IS6f', + 'IS6g', + 'IS7a', + 'IS8a', + 'IS8b', + 'IS9a', + 'IS9b', + 'IS9c', + 'IS9d', + 'O2', + 'P1', + 'P2', + 'P3', + 'P4', + 'P6', + 'P7', + 'P8', + 'UNKNOWN', + 'X1', + 'X2', + 'X3', + 'XXX', + 'Z2', + 'Z3', + 'Z4a', + 'Z4b', + 'Z4c', + 'Z4d', + 'Z4e', + 'Z7', + 'Z9', + ), + palette=[ + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ( + 220, + 20, + 60, + ), + ])