| { |
| "architectures": [ |
| "InkEraserModel" |
| ], |
| "model_type": "unet_plus_plus", |
| "encoder_name": "resnet50", |
| "encoder_weights": "imagenet", |
| "in_channels": 3, |
| "classes": 1, |
| "decoder_attention_type": "scse", |
| "activation": "sigmoid", |
| "framework": "pytorch-lightning", |
| "task": "binary-segmentation", |
| "input_normalization": { |
| "mean": [ |
| 0.485, |
| 0.456, |
| 0.406 |
| ], |
| "std": [ |
| 0.229, |
| 0.224, |
| 0.225 |
| ], |
| "max_pixel_value": 255.0 |
| }, |
| "training_config": { |
| "optimizer": "AdamW", |
| "lr": 3e-05, |
| "weight_decay": 1e-05, |
| "loss": { |
| "dice_weight": 0.8, |
| "bce_weight": 0.2, |
| "bce_pos_weight": 2.0, |
| "dice_from_logits": true |
| }, |
| "metrics": [ |
| "iou_03", |
| "iou_05", |
| "f1_03", |
| "f1_05" |
| ], |
| "scheduler": { |
| "name": "ReduceLROnPlateau", |
| "mode": "max", |
| "monitor": "val_iou_05", |
| "factor": 0.5, |
| "patience": 2, |
| "min_lr": 1e-07 |
| }, |
| "train": { |
| "batch_size": 6, |
| "crop_size": 1024, |
| "accumulate_grad_batches": 4, |
| "precision": "bf16-mixed", |
| "max_epochs": 200, |
| "min_epochs": 5 |
| } |
| }, |
| "inference": { |
| "sliding_window": { |
| "tile": 1024, |
| "stride": 768, |
| "tta_hflip": true |
| } |
| }, |
| "description": "ExamInk-Seg 二值墨迹分割模型（U-Net++ + ResNet50 编码器）。", |
| "license": "MIT" |
| } |
|
|
