config.json

{
  "source_checkpoint": "learning_curve_100pct_20260305_121700_best.ckpt",
  "files": {
    "fp32": "model_fp32.onnx",
    "fp16": "model_fp16.onnx"
  },
  "in_channels": 6,
  "out_channels": 1,
  "resolution": [
    512,
    512
  ],
  "n_angles": 6,
  "rotation_axis": "x",
  "light_source": [
    0.2,
    0.2,
    1.0
  ],
  "colormap": "viridis",
  "additional_face_features": [
    "max_angles"
  ],
  "norm": "minmax",
  "thresholds": [
    0.5,
    0.8
  ],
  "gamma": 0.95,
  "opset": 17,
  "encoder_name": "efficientnet-b5",
  "model_type": "unetplusplus"
}