kneeSeg / RMKV_kneeSeg.json
TommyNgx's picture
Upload folder using huggingface_hub
f18cc1f verified
{
"checkpoint_path": "/projects/BMammo/Knee/Runs/best_models/checkpoint_top1.pth",
"exported_model_path": "/projects/BMammo/Knee/Runs/deploy_export/model_deploy_pickle.pth",
"export_format": "pickle",
"preview_path": "/projects/BMammo/Knee/Runs/deploy_export/random_preview_3.png",
"model_name": "RWKV_UNet",
"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": [
"Pickle export avoids an explicit build_model() call at inference time, but it still requires the same repo modules and custom ops to be importable."
],
"runtime_note": "RWKV models in this project rely on the registered custom WKV operator. Any export format here still needs this repo/runtime available for inference."
}