syding9 commited on
Commit
67d5851
·
verified ·
1 Parent(s): d7ec8fc

Simplify model card

Browse files
Files changed (1) hide show
  1. README.md +0 -34
README.md CHANGED
@@ -50,37 +50,3 @@ model, model_config, training_config, step = load_checkpoint(
50
  ```
51
 
52
  Select `FM_Condition_gradient/latest.pt` for the gradient-loss model.
53
-
54
- ## HiPredict API
55
-
56
- The HiPredict API adapter reconstructs RED4 (`ir2red`), on patches or full strips. Large images are predicted with overlapping tiles.
57
-
58
- ```python
59
- from hipredict.diffusion import build_hipredict_predictor
60
-
61
- predictor, config, step = build_hipredict_predictor(checkpoint, device="cuda")
62
- samples = predictor.load("files/data_record_evals.csv", mode="patches")
63
- predictions = predictor.predict(samples)
64
- ```
65
-
66
- For `red2ir`, use the diffusion inference command with the downloaded checkpoint path:
67
-
68
- ```bash
69
- python -m hipredict.diffusion.inference.inference --ckpt_path /path/to/latest.pt --input_path observation_RED4.npy --red3_path observation_RED3.npy --direction red2ir --output_dir outputs/inference/
70
- ```
71
-
72
- ## Brightness correction
73
-
74
- The default correction uses the RED3 boundary. To use the learned residual brightness head for dataset inference or evaluation, download `brightness/model.pt`, select `--dc_correct head`, and pass its local path as `--head_ckpt`.
75
-
76
- ## Model configuration
77
-
78
- Both FM checkpoints use a 128-channel base width, three residual blocks per level, Sobel edge conditioning, and noise-conditioned flow matching. Their normalization settings are stored in the checkpoint:
79
-
80
- - Center: `0.103687465`
81
- - Scale: `0.04842728`
82
- - Normalized training values clipped to `[-5, 5]`
83
-
84
- HiPredict reads these settings when loading the checkpoint. Evaluation uses original, unclipped physical targets. The checkpoint format is `hipredict.diffusion`, version `1`.
85
-
86
- File SHA-256 digests are recorded in `checksums.json`.
 
50
  ```
51
 
52
  Select `FM_Condition_gradient/latest.pt` for the gradient-loss model.