AURA_Net_oai512.json

{
    "checkpoint_path": "/projects/BMammo/Knee/Runs/oai_512_aura_net/best_models/checkpoint_top1.pth",
    "exported_model_path": "/projects/BMammo/Knee/Runs/oai_512_aura_net/model_deploy.pth",
    "export_format": "checkpoint",
    "preview_path": "/projects/BMammo/Knee/Runs/oai_512_aura_net/random_preview_3.png",
    "model_name": "AURA_Net",
    "img_size": 256,
    "input_channel": 3,
    "num_classes": 1,
    "device": "cuda",
    "threshold": 0.5,
    "samples": [
        {
            "stem": "9671958",
            "image_path": "/projects/BMammo/Knee/data/OAI_kneeFull/images/9671958.jpg",
            "mask_path": "/projects/BMammo/Knee/data/OAI_kneeFull/masks/0/9671958.png"
        },
        {
            "stem": "9597990",
            "image_path": "/projects/BMammo/Knee/data/OAI_kneeFull/images/9597990.jpg",
            "mask_path": "/projects/BMammo/Knee/data/OAI_kneeFull/masks/0/9597990.png"
        },
        {
            "stem": "9440883",
            "image_path": "/projects/BMammo/Knee/data/OAI_kneeFull/images/9440883.jpg",
            "mask_path": "/projects/BMammo/Knee/data/OAI_kneeFull/masks/0/9440883.png"
        }
    ],
    "export_notes": [
        "Compatibility mode: '--export_format pickle' no longer saves a pickled nn.Module and now saves a standard deploy checkpoint '.pth'."
    ]
}