MONAI/breast_density_classification

1{2    "import": [3        "$import glob",4        "$import os",5        "$import torchvision"6    ],7    "bundle_root": ".",8    "model_dir": "$@bundle_root + '/models'",9    "output_dir": "$@bundle_root + '/output'",10    "data": {11        "_target_": "scripts.createList.CreateImageLabelList",12        "filename": "./configs/sample_image_data.json"13    },14    "test_imagelist": "$@data.create_dataset('Test')[0]",15    "test_labellist": "$@data.create_dataset('Test')[1]",16    "dataset": {17        "_target_": "CacheDataset",18        "data": "$[{'image': i, 'label': l} for i, l in zip(@test_imagelist, @test_labellist)]",19        "transform": "@preprocessing",20        "cache_rate": 1,21        "num_workers": 422    },23    "dataloader": {24        "_target_": "DataLoader",25        "dataset": "@dataset",26        "batch_size": 4,27        "shuffle": false,28        "num_workers": 429    },30    "device": "$torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')",31    "network_def": {32        "_target_": "TorchVisionFCModel",33        "model_name": "inception_v3",34        "num_classes": 4,35        "pool": null,36        "use_conv": false,37        "bias": true,38        "pretrained": true39    },40    "network": "$@network_def.to(@device)",41    "preprocessing": {42        "_target_": "Compose",43        "transforms": [44            {45                "_target_": "LoadImaged",46                "keys": "image"47            },48            {49                "_target_": "EnsureChannelFirstd",50                "keys": "image",51                "channel_dim": 252            },53            {54                "_target_": "ScaleIntensityd",55                "keys": "image",56                "minv": 0.0,57                "maxv": 1.058            },59            {60                "_target_": "Resized",61                "keys": "image",62                "spatial_size": [63                    299,64                    29965                ]66            }67        ]68    },69    "inferer": {70        "_target_": "SimpleInferer"71    },72    "postprocessing": {73        "_target_": "Compose",74        "transforms": [75            {76                "_target_": "Activationsd",77                "keys": "pred",78                "sigmoid": true79            }80        ]81    },82    "handlers": [83        {84            "_target_": "CheckpointLoader",85            "load_path": "$@model_dir + '/model.pt'",86            "load_dict": {87                "model": "@network"88            }89        },90        {91            "_target_": "StatsHandler",92            "iteration_log": false,93            "output_transform": "$lambda x: None"94        },95        {96            "_target_": "ClassificationSaver",97            "output_dir": "@output_dir",98            "batch_transform": "$lambda x: [xx['image'].meta for xx in x]",99            "output_transform": "$monai.handlers.from_engine(['pred'])"100        }101    ],102    "evaluator": {103        "_target_": "SupervisedEvaluator",104        "device": "@device",105        "val_data_loader": "@dataloader",106        "network": "@network",107        "inferer": "@inferer",108        "postprocessing": "@postprocessing",109        "val_handlers": "@handlers",110        "amp": true111    },112    "evaluating": [113        "$setattr(torch.backends.cudnn, 'benchmark', True)",114        "$@evaluator.run()"115    ]116}117