diff --git a/config/psenet/psenet_r50_ic15_1280.py b/config/psenet/psenet_r50_ic15_1280.py deleted file mode 100644 index 0388749..0000000 --- a/config/psenet/psenet_r50_ic15_1280.py +++ /dev/null @@ -1,58 +0,0 @@ -model = dict( - type='PSENet', - backbone=dict( - type='resnet50', - pretrained=True - ), - neck=dict( - type='FPN', - in_channels=(256, 512, 1024, 2048), - out_channels=128 - ), - detection_head=dict( - type='PSENet_Head', - in_channels=1024, - hidden_dim=256, - num_classes=7, - loss_text=dict( - type='DiceLoss', - loss_weight=0.7 - ), - loss_kernel=dict( - type='DiceLoss', - loss_weight=0.3 - ) - ) -) -data = dict( - batch_size=16, - train=dict( - type='PSENET_IC15', - split='train', - is_transform=True, - img_size=736, - short_size=1280, - kernel_num=7, - min_scale=0.4, - read_type='cv2' - ), - test=dict( - type='PSENET_IC15', - split='test', - short_size=1280, - read_type='cv2' - ) -) -train_cfg = dict( - lr=1e-3, - schedule=(200, 400,), - epoch=600, - optimizer='SGD' -) -test_cfg = dict( - min_score=0.85, - min_area=16, - kernel_num=7, - bbox_type='rect', - result_path='outputs/submit_ic15.zip' -) diff --git a/vis_test/vis_kernels.png b/vis_test/vis_kernels.png deleted file mode 100644 index 6d9f98d..0000000 Binary files a/vis_test/vis_kernels.png and /dev/null differ