KANResDiff/metadata_manager.py at main · PerceptionComputingLab/KANResDiff · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
# Harbin Institute of Technology Bachelor Thesis
# Author: HIT Michael_Bryant
# Mail: 1137892110@qq.com
from config import *

cfg = load_global_config()

"""
Store all dataset/task relevant meta data here for passing them to the training script.
"""


def get_meta(task):
    if task == "LIDC":
        meta = {
            "description": "LIDC Lung Module Dataset (subset with 4 annotations)",
            "channels": 1,
            # "all_data_path": "/Share8/Michael_Bryant/data_LIDC/",
            "all_data_path": cfg_get(cfg, ['data', 'tasks', task], default=""),
            "masking_threshold": 0.5,
            "image_size": 128,
            "admissible_size": 128,
            "output_size": 128,
            "directory_name": "LIDC",
            "raters": 4,
            "num_filters": [32, 64, 128, 192],
            # 'lossfunction': define lossfunction here
        }
        return meta
    if task == "isic3_style_concat":
        meta = {
            "description": "ISIC Skin Lesion Dataset with same split as style subsets ",
            "channels": 3,
            "all_data_path": cfg_get(cfg, ['data', 'tasks', task], default=""),
            "masking_threshold": 0.5,
            "image_size": 256,
            "admissible_size": 340,
            "output_size": 252,
            "directory_name": "isic3",
            "raters": 3,
            "num_filters": [32, 64, 128, 192],
            # 'lossfunction': define lossfunction here
        }
        return meta