Upload model parameters and training configs

2025-10-18/model_state_dict.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0486b444a7563eb2d52b1120e13d1666fa301705a65ef352cd41c0cb4cfae081
+size 160284421

2025-10-18/settings.json

@@ -0,0 +1,187 @@
+{
+    "experiment_name": "Modified_U-Net_2025-10-18_14h07m24s",
+    "dataset_filename": "retraining.xlsx",
+    "seed": 64,
+    "model_name": "ModifiedUNet",
+    "compile_model": false,
+    "checkpoint_path": false,
+    "model": {
+        "input_channels": 3,
+        "filters": 32,
+        "downsample_factors": [
+            2,
+            4,
+            4,
+            4,
+            4
+        ],
+        "residual_connection": false,
+        "attach_tissue_decoder": true,
+        "attach_pen_decoder": true,
+        "attach_distance_decoder": true
+    },
+    "training": {
+        "learning_rate": 0.0003,
+        "iterations": 100000,
+        "iterations_per_decay": 20000,
+        "decay_factor": 0.5,
+        "iterations_per_checkpoint": 500,
+        "iterations_per_update": 5,
+        "loss": {
+            "loss_weights": {
+                "tversky": 1.0,
+                "focal": 10.0,
+                "MSE dist": 1.0,
+                "MSE grad dist": 1.0
+            },
+            "class_weights": {
+                "tissue": 1.0,
+                "pen": 1.0
+            },
+            "fp_weight": 0.5,
+            "fn_weight": 0.5,
+            "gamma": 0,
+            "smooth_nom": 1,
+            "smooth_denom": 1
+        }
+    },
+    "dataloader": {
+        "num_workers": 4,
+        "pin_memory": true,
+        "batch_size": 1,
+        "image_shape": null,
+        "max_image_shape": [
+            3072,
+            3072
+        ]
+    },
+    "augmentation": {
+        "RandomRotate90": {
+            "p": 0.25
+        },
+        "Affine": {
+            "p": 0.5,
+            "translate_px": {
+                "x": [
+                    -256,
+                    256
+                ],
+                "y": [
+                    -256,
+                    256
+                ]
+            },
+            "scale": [
+                0.95,
+                1.05
+            ],
+            "rotate": [
+                -180,
+                180
+            ]
+        },
+        "HorizontalFlip": {
+            "p": 0.5
+        },
+        "VerticalFlip": {
+            "p": 0.5
+        },
+        "HueSaturationValue": {
+            "p": 0.5,
+            "hue_shift_limit": 25,
+            "sat_shift_limit": 5,
+            "val_shift_limit": 0
+        },
+        "HueSaturationValue pen": {
+            "p": 0.25,
+            "hue_shift_limit": 10,
+            "sat_shift_limit": 5,
+            "val_shift_limit": 0
+        },
+        "HueSaturationValue tissue": {
+            "p": 0.25,
+            "hue_shift_limit": 10,
+            "sat_shift_limit": 5,
+            "val_shift_limit": 0
+        },
+        "HueSaturationValue non-tissue": {
+            "p": 0.0,
+            "hue_shift_limit": 0,
+            "sat_shift_limit": 0,
+            "val_shift_limit": 0
+        },
+        "HueSaturationValue background": {
+            "p": 0.25,
+            "hue_shift_limit": 10,
+            "sat_shift_limit": 5,
+            "val_shift_limit": 0
+        },
+        "RandomBrightnessContrast": {
+            "p": 0.25,
+            "brightness_limit": 0.2,
+            "contrast_limit": 0.2
+        },
+        "RandomGamma": {
+            "p": 0.25,
+            "gamma_limit": [
+                67,
+                150
+            ]
+        },
+        "GaussNoise": {
+            "p": 0.2,
+            "var_limit": [
+                0,
+                25
+            ]
+        },
+        "GaussianBlur": {
+            "p": 0.2,
+            "sigma_limit": [
+                1e-05,
+                2
+            ]
+        },
+        "JPEGCompression": {
+            "p": 0.2,
+            "quality_limit": [
+                25,
+                100
+            ]
+        },
+        "Padding": [
+            {
+                "mode": "reflect",
+                "p": 0.9
+            },
+            {
+                "mode": "constant",
+                "p": 0.0,
+                "value": 0
+            },
+            {
+                "mode": "constant",
+                "p": 0.1,
+                "value": 255
+            }
+        ],
+        "PenMarkings": {
+            "p": 0.1,
+            "N": [
+                2,
+                6
+            ]
+        }
+    },
+    "hyperparameters": {
+        "tissue_threshold": 0.5,
+        "pen_marking_threshold": 0.8,
+        "pixels_per_bin": 20,
+        "sigma": 3.0,
+        "filter_size": 15,
+        "percentile": 98,
+        "padding_mode": "constant",
+        "padding_value": 1.0,
+        "distance_factor": 100
+    }
+}