diff options
| author | skal <pascal.massimino@gmail.com> | 2026-03-22 11:16:51 +0100 |
|---|---|---|
| committer | skal <pascal.massimino@gmail.com> | 2026-03-22 11:16:51 +0100 |
| commit | bce23b0deb33cadd54d01765576965c3cfb96d69 (patch) | |
| tree | d5d7e6ebde634e99eb812b3dbd5408bb5115b295 /cnn_v3 | |
| parent | e6d847b711d68c9eeaa589e632b8f6edf9fec2f2 (diff) | |
docs(cnn_v3): add uv inline deps to train_cnn_v3.py + HOW_TO_CNN note
handoff(Gemini): train_cnn_v3.py now has uv script metadata block
(torch, torchvision, numpy, pillow, opencv-python). HOW_TO_CNN §2
Prerequisites updated with uv quick-start alternative.
Diffstat (limited to 'cnn_v3')
| -rw-r--r-- | cnn_v3/docs/HOW_TO_CNN.md | 7 | ||||
| -rw-r--r-- | cnn_v3/training/train_cnn_v3.py | 4 |
2 files changed, 11 insertions, 0 deletions
diff --git a/cnn_v3/docs/HOW_TO_CNN.md b/cnn_v3/docs/HOW_TO_CNN.md index 4e64d23..020f79c 100644 --- a/cnn_v3/docs/HOW_TO_CNN.md +++ b/cnn_v3/docs/HOW_TO_CNN.md @@ -289,6 +289,13 @@ pip install torch torchvision pillow numpy opencv-python cd cnn_v3/training ``` +**With `uv` (no pip needed):** dependencies are declared inline in `train_cnn_v3.py` +and installed automatically on first run: +```bash +cd cnn_v3/training +uv run train_cnn_v3.py --input dataset/ --epochs 1 --patch-size 32 --detector random +``` + ### Quick-start commands **Smoke test — 1 epoch, validates end-to-end without GPU:** diff --git a/cnn_v3/training/train_cnn_v3.py b/cnn_v3/training/train_cnn_v3.py index 6da9abb..083efb0 100644 --- a/cnn_v3/training/train_cnn_v3.py +++ b/cnn_v3/training/train_cnn_v3.py @@ -1,4 +1,8 @@ #!/usr/bin/env python3 +# /// script +# requires-python = ">=3.10" +# dependencies = ["torch", "torchvision", "numpy", "pillow", "opencv-python"] +# /// """CNN v3 Training Script — U-Net + FiLM Architecture: |