test

Author: qilei123

configs/apps/trans_drone/full_segformer_mit-b5_640x640_160k_td.py

@@ -1,6 +1,6 @@
 norm_cfg = dict(type='SyncBN', requires_grad=True)
-num_classes = 4
-dataset_type = 'TDDataset4'
+num_classes = 3
+dataset_type = 'TDDataset'
 data_root = 'data/td/'
 img_norm_cfg = dict(
     mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True)
@@ -124,4 +124,4 @@
 runner = dict(type='IterBasedRunner', max_iters=64000)
 checkpoint_config = dict(by_epoch=False, interval=6400)
 evaluation = dict(interval=6400, metric='mIoU', pre_eval=True)
-work_dir = data_root+'work_dirs/full_segformer_mit-b5_640x640_160k_td_bg/'
+work_dir = data_root+'work_dirs/full_segformer_mit-b5_640x640_160k_td_nbg/'