Simplify model card
Browse files
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.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|