monai
medical
File size: 3,602 Bytes
a23faaa
 
9754d0b
a23faaa
9754d0b
b27ec38
2ab6e57
fbd9231
b725cb6
4e1290e
47c6cd2
31d89d2
28f3221
df63360
460afce
50252f2
715e62e
d38f56f
9a5a418
e4ba616
01573e4
a23faaa
 
 
 
9754d0b
31d89d2
 
a23faaa
 
 
 
28f3221
 
a23faaa
 
 
 
 
 
 
 
 
fbd9231
a23faaa
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
{
    "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
    "version": "0.4.4",
    "changelog": {
        "0.4.4": "fix the wrong GPU index issue of multi-node",
        "0.4.3": "add dataset dir example",
        "0.4.2": "update ONNX-TensorRT descriptions",
        "0.4.1": "update the model weights with the deterministic training",
        "0.4.0": "add the ONNX-TensorRT way of model conversion",
        "0.3.9": "fix mgpu finalize issue",
        "0.3.8": "enable deterministic training",
        "0.3.7": "adapt to BundleWorkflow interface",
        "0.3.6": "add name tag",
        "0.3.5": "fix a comment issue in the data_process script",
        "0.3.4": "add note for multi-gpu training with example dataset",
        "0.3.3": "enhance data preprocess script and readme file",
        "0.3.2": "restructure readme to match updated template",
        "0.3.1": "add workflow, train loss and validation accuracy figures",
        "0.3.0": "update dataset processing",
        "0.2.2": "update to use monai 1.0.1",
        "0.2.1": "enhance readme on commands example",
        "0.2.0": "update license files",
        "0.1.0": "complete the first version model package",
        "0.0.1": "initialize the model package structure"
    },
    "monai_version": "1.2.0",
    "pytorch_version": "1.13.1",
    "numpy_version": "1.22.2",
    "optional_packages_version": {
        "nibabel": "4.0.1",
        "pytorch-ignite": "0.4.9"
    },
    "name": "Endoscopic inbody classification",
    "task": "Endoscopic inbody classification",
    "description": "A pre-trained binary classification model for endoscopic inbody classification task",
    "authors": "NVIDIA DLMED team",
    "copyright": "Copyright (c) 2021-2022, NVIDIA CORPORATION",
    "data_source": "private dataset",
    "data_type": "RGB",
    "image_classes": "three channel data, intensity [0-255]",
    "label_classes": "0: inbody, 1: outbody",
    "pred_classes": "vector whose length equals to 2, [1,0] means in body, [0,1] means out body",
    "eval_metrics": {
        "accuracy": 0.99
    },
    "references": [
        "J. Hu, L. Shen and G. Sun, Squeeze-and-Excitation Networks, 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2018, pp. 7132-7141. https://arxiv.org/pdf/1709.01507.pdf"
    ],
    "network_data_format": {
        "inputs": {
            "image": {
                "type": "magnitude",
                "format": "RGB",
                "modality": "regular",
                "num_channels": 3,
                "spatial_shape": [
                    256,
                    256
                ],
                "dtype": "float32",
                "value_range": [
                    0,
                    1
                ],
                "is_patch_data": false,
                "channel_def": {
                    "0": "R",
                    "1": "G",
                    "2": "B"
                }
            }
        },
        "outputs": {
            "pred": {
                "type": "probabilities",
                "format": "classes",
                "num_channels": 2,
                "spatial_shape": [
                    1,
                    2
                ],
                "dtype": "float32",
                "value_range": [
                    0,
                    1
                ],
                "is_patch_data": false,
                "channel_def": {
                    "0": "in body",
                    "1": "out body"
                }
            }
        }
    }
}