From fcd1ceb2168de76659091208f911d2c4d0c59ad7 Mon Sep 17 00:00:00 2001 From: skal Date: Sun, 22 Mar 2026 07:33:28 +0100 Subject: feat(cnn_v3): gen_sample tool + 7 simple training samples MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - pack_photo_sample.py: --target now required (no albedo fallback) - gen_sample: bash wrapper with positional args (input target output_dir) - input/photo7.jpg: copy of photo2 (second style target) - target_1: photo2_1_out→photo2_out, photo2_2_out→photo7_out - dataset/simple/sample_001..007: 7 packed photo/target pairs handoff(Gemini): training data ready; next step is train_cnn_v3.py run --- .../training/dataset/simple/sample_001/albedo.png | Bin 0 -> 22390911 bytes .../training/dataset/simple/sample_001/depth.png | Bin 0 -> 23365 bytes .../training/dataset/simple/sample_001/matid.png | Bin 0 -> 11724 bytes .../training/dataset/simple/sample_001/normal.png | Bin 0 -> 44260 bytes .../training/dataset/simple/sample_001/shadow.png | Bin 0 -> 20443 bytes .../training/dataset/simple/sample_001/target.png | Bin 0 -> 4363727 bytes .../training/dataset/simple/sample_001/transp.png | Bin 0 -> 11724 bytes .../training/dataset/simple/sample_002/albedo.png | Bin 0 -> 14366815 bytes .../training/dataset/simple/sample_002/depth.png | Bin 0 -> 23365 bytes .../training/dataset/simple/sample_002/matid.png | Bin 0 -> 11724 bytes .../training/dataset/simple/sample_002/normal.png | Bin 0 -> 44260 bytes .../training/dataset/simple/sample_002/shadow.png | Bin 0 -> 20443 bytes .../training/dataset/simple/sample_002/target.png | Bin 0 -> 3079705 bytes .../training/dataset/simple/sample_002/transp.png | Bin 0 -> 11724 bytes .../training/dataset/simple/sample_003/albedo.png | Bin 0 -> 7252297 bytes .../training/dataset/simple/sample_003/depth.png | Bin 0 -> 14234 bytes .../training/dataset/simple/sample_003/matid.png | Bin 0 -> 7151 bytes .../training/dataset/simple/sample_003/normal.png | Bin 0 -> 25739 bytes .../training/dataset/simple/sample_003/shadow.png | Bin 0 -> 11255 bytes .../training/dataset/simple/sample_003/target.png | Bin 0 -> 2436109 bytes .../training/dataset/simple/sample_003/transp.png | Bin 0 -> 7151 bytes .../training/dataset/simple/sample_004/albedo.png | Bin 0 -> 1245962 bytes .../training/dataset/simple/sample_004/depth.png | Bin 0 -> 7389 bytes .../training/dataset/simple/sample_004/matid.png | Bin 0 -> 3735 bytes .../training/dataset/simple/sample_004/normal.png | Bin 0 -> 15342 bytes .../training/dataset/simple/sample_004/shadow.png | Bin 0 -> 7738 bytes .../training/dataset/simple/sample_004/target.png | Bin 0 -> 731778 bytes .../training/dataset/simple/sample_004/transp.png | Bin 0 -> 3735 bytes .../training/dataset/simple/sample_005/albedo.png | Bin 0 -> 1187567 bytes .../training/dataset/simple/sample_005/depth.png | Bin 0 -> 1632 bytes .../training/dataset/simple/sample_005/matid.png | Bin 0 -> 857 bytes .../training/dataset/simple/sample_005/normal.png | Bin 0 -> 4590 bytes .../training/dataset/simple/sample_005/shadow.png | Bin 0 -> 3137 bytes .../training/dataset/simple/sample_005/target.png | Bin 0 -> 3140126 bytes .../training/dataset/simple/sample_005/transp.png | Bin 0 -> 857 bytes .../training/dataset/simple/sample_006/albedo.png | Bin 0 -> 2363392 bytes .../training/dataset/simple/sample_006/depth.png | Bin 0 -> 2988 bytes .../training/dataset/simple/sample_006/matid.png | Bin 0 -> 1534 bytes .../training/dataset/simple/sample_006/normal.png | Bin 0 -> 7774 bytes .../training/dataset/simple/sample_006/shadow.png | Bin 0 -> 4399 bytes .../training/dataset/simple/sample_006/target.png | Bin 0 -> 3153440 bytes .../training/dataset/simple/sample_006/transp.png | Bin 0 -> 1534 bytes .../training/dataset/simple/sample_007/albedo.png | Bin 0 -> 14366815 bytes .../training/dataset/simple/sample_007/depth.png | Bin 0 -> 23365 bytes .../training/dataset/simple/sample_007/matid.png | Bin 0 -> 11724 bytes .../training/dataset/simple/sample_007/normal.png | Bin 0 -> 44260 bytes .../training/dataset/simple/sample_007/shadow.png | Bin 0 -> 20443 bytes .../training/dataset/simple/sample_007/target.png | Bin 0 -> 3084795 bytes .../training/dataset/simple/sample_007/transp.png | Bin 0 -> 11724 bytes cnn_v3/training/gen_sample | 21 +++++++++++++++++++++ cnn_v3/training/input/photo7.jpg | Bin 0 -> 3328126 bytes cnn_v3/training/pack_photo_sample.py | 17 ++++++++++------- cnn_v3/training/target_1/photo2_1_out.png | Bin 2875939 -> 0 bytes cnn_v3/training/target_1/photo2_2_out.png | Bin 2865857 -> 0 bytes cnn_v3/training/target_1/photo2_out.png | Bin 0 -> 2875939 bytes cnn_v3/training/target_1/photo7_out.png | Bin 0 -> 2865857 bytes 56 files changed, 31 insertions(+), 7 deletions(-) create mode 100644 cnn_v3/training/dataset/simple/sample_001/albedo.png create mode 100644 cnn_v3/training/dataset/simple/sample_001/depth.png create mode 100644 cnn_v3/training/dataset/simple/sample_001/matid.png create mode 100644 cnn_v3/training/dataset/simple/sample_001/normal.png create mode 100644 cnn_v3/training/dataset/simple/sample_001/shadow.png create mode 100644 cnn_v3/training/dataset/simple/sample_001/target.png create mode 100644 cnn_v3/training/dataset/simple/sample_001/transp.png create mode 100644 cnn_v3/training/dataset/simple/sample_002/albedo.png create mode 100644 cnn_v3/training/dataset/simple/sample_002/depth.png create mode 100644 cnn_v3/training/dataset/simple/sample_002/matid.png create mode 100644 cnn_v3/training/dataset/simple/sample_002/normal.png create mode 100644 cnn_v3/training/dataset/simple/sample_002/shadow.png create mode 100644 cnn_v3/training/dataset/simple/sample_002/target.png create mode 100644 cnn_v3/training/dataset/simple/sample_002/transp.png create mode 100644 cnn_v3/training/dataset/simple/sample_003/albedo.png create mode 100644 cnn_v3/training/dataset/simple/sample_003/depth.png create mode 100644 cnn_v3/training/dataset/simple/sample_003/matid.png create mode 100644 cnn_v3/training/dataset/simple/sample_003/normal.png create mode 100644 cnn_v3/training/dataset/simple/sample_003/shadow.png create mode 100644 cnn_v3/training/dataset/simple/sample_003/target.png create mode 100644 cnn_v3/training/dataset/simple/sample_003/transp.png create mode 100644 cnn_v3/training/dataset/simple/sample_004/albedo.png create mode 100644 cnn_v3/training/dataset/simple/sample_004/depth.png create mode 100644 cnn_v3/training/dataset/simple/sample_004/matid.png create mode 100644 cnn_v3/training/dataset/simple/sample_004/normal.png create mode 100644 cnn_v3/training/dataset/simple/sample_004/shadow.png create mode 100644 cnn_v3/training/dataset/simple/sample_004/target.png create mode 100644 cnn_v3/training/dataset/simple/sample_004/transp.png create mode 100644 cnn_v3/training/dataset/simple/sample_005/albedo.png create mode 100644 cnn_v3/training/dataset/simple/sample_005/depth.png create mode 100644 cnn_v3/training/dataset/simple/sample_005/matid.png create mode 100644 cnn_v3/training/dataset/simple/sample_005/normal.png create mode 100644 cnn_v3/training/dataset/simple/sample_005/shadow.png create mode 100644 cnn_v3/training/dataset/simple/sample_005/target.png create mode 100644 cnn_v3/training/dataset/simple/sample_005/transp.png create mode 100644 cnn_v3/training/dataset/simple/sample_006/albedo.png create mode 100644 cnn_v3/training/dataset/simple/sample_006/depth.png create mode 100644 cnn_v3/training/dataset/simple/sample_006/matid.png create mode 100644 cnn_v3/training/dataset/simple/sample_006/normal.png create mode 100644 cnn_v3/training/dataset/simple/sample_006/shadow.png create mode 100644 cnn_v3/training/dataset/simple/sample_006/target.png create mode 100644 cnn_v3/training/dataset/simple/sample_006/transp.png create mode 100644 cnn_v3/training/dataset/simple/sample_007/albedo.png create mode 100644 cnn_v3/training/dataset/simple/sample_007/depth.png create mode 100644 cnn_v3/training/dataset/simple/sample_007/matid.png create mode 100644 cnn_v3/training/dataset/simple/sample_007/normal.png create mode 100644 cnn_v3/training/dataset/simple/sample_007/shadow.png create mode 100644 cnn_v3/training/dataset/simple/sample_007/target.png create mode 100644 cnn_v3/training/dataset/simple/sample_007/transp.png create mode 100755 cnn_v3/training/gen_sample create mode 100644 cnn_v3/training/input/photo7.jpg delete mode 100644 cnn_v3/training/target_1/photo2_1_out.png delete mode 100644 cnn_v3/training/target_1/photo2_2_out.png create mode 100644 cnn_v3/training/target_1/photo2_out.png create mode 100644 cnn_v3/training/target_1/photo7_out.png diff --git a/cnn_v3/training/dataset/simple/sample_001/albedo.png b/cnn_v3/training/dataset/simple/sample_001/albedo.png new file mode 100644 index 0000000..68220b3 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_001/albedo.png differ diff --git a/cnn_v3/training/dataset/simple/sample_001/depth.png b/cnn_v3/training/dataset/simple/sample_001/depth.png new file mode 100644 index 0000000..5708605 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_001/depth.png differ diff --git a/cnn_v3/training/dataset/simple/sample_001/matid.png b/cnn_v3/training/dataset/simple/sample_001/matid.png new file mode 100644 index 0000000..2561108 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_001/matid.png differ diff --git a/cnn_v3/training/dataset/simple/sample_001/normal.png b/cnn_v3/training/dataset/simple/sample_001/normal.png new file mode 100644 index 0000000..489f629 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_001/normal.png differ diff --git a/cnn_v3/training/dataset/simple/sample_001/shadow.png b/cnn_v3/training/dataset/simple/sample_001/shadow.png new file mode 100644 index 0000000..f356fb2 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_001/shadow.png differ diff --git a/cnn_v3/training/dataset/simple/sample_001/target.png b/cnn_v3/training/dataset/simple/sample_001/target.png new file mode 100644 index 0000000..268e167 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_001/target.png differ diff --git a/cnn_v3/training/dataset/simple/sample_001/transp.png b/cnn_v3/training/dataset/simple/sample_001/transp.png new file mode 100644 index 0000000..2561108 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_001/transp.png differ diff --git a/cnn_v3/training/dataset/simple/sample_002/albedo.png b/cnn_v3/training/dataset/simple/sample_002/albedo.png new file mode 100644 index 0000000..80f4f1c Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_002/albedo.png differ diff --git a/cnn_v3/training/dataset/simple/sample_002/depth.png b/cnn_v3/training/dataset/simple/sample_002/depth.png new file mode 100644 index 0000000..5708605 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_002/depth.png differ diff --git a/cnn_v3/training/dataset/simple/sample_002/matid.png b/cnn_v3/training/dataset/simple/sample_002/matid.png new file mode 100644 index 0000000..2561108 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_002/matid.png differ diff --git a/cnn_v3/training/dataset/simple/sample_002/normal.png b/cnn_v3/training/dataset/simple/sample_002/normal.png new file mode 100644 index 0000000..489f629 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_002/normal.png differ diff --git a/cnn_v3/training/dataset/simple/sample_002/shadow.png b/cnn_v3/training/dataset/simple/sample_002/shadow.png new file mode 100644 index 0000000..f356fb2 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_002/shadow.png differ diff --git a/cnn_v3/training/dataset/simple/sample_002/target.png b/cnn_v3/training/dataset/simple/sample_002/target.png new file mode 100644 index 0000000..d1d7192 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_002/target.png differ diff --git a/cnn_v3/training/dataset/simple/sample_002/transp.png b/cnn_v3/training/dataset/simple/sample_002/transp.png new file mode 100644 index 0000000..2561108 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_002/transp.png differ diff --git a/cnn_v3/training/dataset/simple/sample_003/albedo.png b/cnn_v3/training/dataset/simple/sample_003/albedo.png new file mode 100644 index 0000000..fe714bc Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_003/albedo.png differ diff --git a/cnn_v3/training/dataset/simple/sample_003/depth.png b/cnn_v3/training/dataset/simple/sample_003/depth.png new file mode 100644 index 0000000..eac97a6 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_003/depth.png differ diff --git a/cnn_v3/training/dataset/simple/sample_003/matid.png b/cnn_v3/training/dataset/simple/sample_003/matid.png new file mode 100644 index 0000000..60a5b09 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_003/matid.png differ diff --git a/cnn_v3/training/dataset/simple/sample_003/normal.png b/cnn_v3/training/dataset/simple/sample_003/normal.png new file mode 100644 index 0000000..36dd1fd Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_003/normal.png differ diff --git a/cnn_v3/training/dataset/simple/sample_003/shadow.png b/cnn_v3/training/dataset/simple/sample_003/shadow.png new file mode 100644 index 0000000..c943f56 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_003/shadow.png differ diff --git a/cnn_v3/training/dataset/simple/sample_003/target.png b/cnn_v3/training/dataset/simple/sample_003/target.png new file mode 100644 index 0000000..56205ca Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_003/target.png differ diff --git a/cnn_v3/training/dataset/simple/sample_003/transp.png b/cnn_v3/training/dataset/simple/sample_003/transp.png new file mode 100644 index 0000000..60a5b09 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_003/transp.png differ diff --git a/cnn_v3/training/dataset/simple/sample_004/albedo.png b/cnn_v3/training/dataset/simple/sample_004/albedo.png new file mode 100644 index 0000000..a1b817d Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_004/albedo.png differ diff --git a/cnn_v3/training/dataset/simple/sample_004/depth.png b/cnn_v3/training/dataset/simple/sample_004/depth.png new file mode 100644 index 0000000..083c4a3 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_004/depth.png differ diff --git a/cnn_v3/training/dataset/simple/sample_004/matid.png b/cnn_v3/training/dataset/simple/sample_004/matid.png new file mode 100644 index 0000000..f14721f Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_004/matid.png differ diff --git a/cnn_v3/training/dataset/simple/sample_004/normal.png b/cnn_v3/training/dataset/simple/sample_004/normal.png new file mode 100644 index 0000000..0694ac1 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_004/normal.png differ diff --git a/cnn_v3/training/dataset/simple/sample_004/shadow.png b/cnn_v3/training/dataset/simple/sample_004/shadow.png new file mode 100644 index 0000000..9b6a5e0 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_004/shadow.png differ diff --git a/cnn_v3/training/dataset/simple/sample_004/target.png b/cnn_v3/training/dataset/simple/sample_004/target.png new file mode 100644 index 0000000..019d7a8 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_004/target.png differ diff --git a/cnn_v3/training/dataset/simple/sample_004/transp.png b/cnn_v3/training/dataset/simple/sample_004/transp.png new file mode 100644 index 0000000..f14721f Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_004/transp.png differ diff --git a/cnn_v3/training/dataset/simple/sample_005/albedo.png b/cnn_v3/training/dataset/simple/sample_005/albedo.png new file mode 100644 index 0000000..5dfada7 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_005/albedo.png differ diff --git a/cnn_v3/training/dataset/simple/sample_005/depth.png b/cnn_v3/training/dataset/simple/sample_005/depth.png new file mode 100644 index 0000000..db65c11 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_005/depth.png differ diff --git a/cnn_v3/training/dataset/simple/sample_005/matid.png b/cnn_v3/training/dataset/simple/sample_005/matid.png new file mode 100644 index 0000000..f098c4d Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_005/matid.png differ diff --git a/cnn_v3/training/dataset/simple/sample_005/normal.png b/cnn_v3/training/dataset/simple/sample_005/normal.png new file mode 100644 index 0000000..40305ec Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_005/normal.png differ diff --git a/cnn_v3/training/dataset/simple/sample_005/shadow.png b/cnn_v3/training/dataset/simple/sample_005/shadow.png new file mode 100644 index 0000000..8e270d3 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_005/shadow.png differ diff --git a/cnn_v3/training/dataset/simple/sample_005/target.png b/cnn_v3/training/dataset/simple/sample_005/target.png new file mode 100644 index 0000000..41ead45 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_005/target.png differ diff --git a/cnn_v3/training/dataset/simple/sample_005/transp.png b/cnn_v3/training/dataset/simple/sample_005/transp.png new file mode 100644 index 0000000..f098c4d Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_005/transp.png differ diff --git a/cnn_v3/training/dataset/simple/sample_006/albedo.png b/cnn_v3/training/dataset/simple/sample_006/albedo.png new file mode 100644 index 0000000..6fd43d4 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_006/albedo.png differ diff --git a/cnn_v3/training/dataset/simple/sample_006/depth.png b/cnn_v3/training/dataset/simple/sample_006/depth.png new file mode 100644 index 0000000..4ccb0bc Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_006/depth.png differ diff --git a/cnn_v3/training/dataset/simple/sample_006/matid.png b/cnn_v3/training/dataset/simple/sample_006/matid.png new file mode 100644 index 0000000..32e1b1c Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_006/matid.png differ diff --git a/cnn_v3/training/dataset/simple/sample_006/normal.png b/cnn_v3/training/dataset/simple/sample_006/normal.png new file mode 100644 index 0000000..e2623bd Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_006/normal.png differ diff --git a/cnn_v3/training/dataset/simple/sample_006/shadow.png b/cnn_v3/training/dataset/simple/sample_006/shadow.png new file mode 100644 index 0000000..880ccf9 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_006/shadow.png differ diff --git a/cnn_v3/training/dataset/simple/sample_006/target.png b/cnn_v3/training/dataset/simple/sample_006/target.png new file mode 100644 index 0000000..383e2d3 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_006/target.png differ diff --git a/cnn_v3/training/dataset/simple/sample_006/transp.png b/cnn_v3/training/dataset/simple/sample_006/transp.png new file mode 100644 index 0000000..32e1b1c Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_006/transp.png differ diff --git a/cnn_v3/training/dataset/simple/sample_007/albedo.png b/cnn_v3/training/dataset/simple/sample_007/albedo.png new file mode 100644 index 0000000..80f4f1c Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_007/albedo.png differ diff --git a/cnn_v3/training/dataset/simple/sample_007/depth.png b/cnn_v3/training/dataset/simple/sample_007/depth.png new file mode 100644 index 0000000..5708605 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_007/depth.png differ diff --git a/cnn_v3/training/dataset/simple/sample_007/matid.png b/cnn_v3/training/dataset/simple/sample_007/matid.png new file mode 100644 index 0000000..2561108 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_007/matid.png differ diff --git a/cnn_v3/training/dataset/simple/sample_007/normal.png b/cnn_v3/training/dataset/simple/sample_007/normal.png new file mode 100644 index 0000000..489f629 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_007/normal.png differ diff --git a/cnn_v3/training/dataset/simple/sample_007/shadow.png b/cnn_v3/training/dataset/simple/sample_007/shadow.png new file mode 100644 index 0000000..f356fb2 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_007/shadow.png differ diff --git a/cnn_v3/training/dataset/simple/sample_007/target.png b/cnn_v3/training/dataset/simple/sample_007/target.png new file mode 100644 index 0000000..ea276e4 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_007/target.png differ diff --git a/cnn_v3/training/dataset/simple/sample_007/transp.png b/cnn_v3/training/dataset/simple/sample_007/transp.png new file mode 100644 index 0000000..2561108 Binary files /dev/null and b/cnn_v3/training/dataset/simple/sample_007/transp.png differ diff --git a/cnn_v3/training/gen_sample b/cnn_v3/training/gen_sample new file mode 100755 index 0000000..21519f1 --- /dev/null +++ b/cnn_v3/training/gen_sample @@ -0,0 +1,21 @@ +#!/usr/bin/env bash +# gen_sample — pack an input/target photo pair into a CNN v3 sample directory. +# +# Usage: +# gen_sample +# +# Example: +# gen_sample input/photo1.jpg target_1/photo1_out.png dataset/simple/sample_001 + +set -euo pipefail + +if [ $# -ne 3 ]; then + echo "Usage: gen_sample " >&2 + exit 1 +fi + +SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)" +python3 "$SCRIPT_DIR/pack_photo_sample.py" \ + --photo "$1" \ + --target "$2" \ + --output "$3" diff --git a/cnn_v3/training/input/photo7.jpg b/cnn_v3/training/input/photo7.jpg new file mode 100644 index 0000000..a662fa8 Binary files /dev/null and b/cnn_v3/training/input/photo7.jpg differ diff --git a/cnn_v3/training/pack_photo_sample.py b/cnn_v3/training/pack_photo_sample.py index b2943fb..ca80301 100644 --- a/cnn_v3/training/pack_photo_sample.py +++ b/cnn_v3/training/pack_photo_sample.py @@ -18,7 +18,8 @@ mip1 and mip2 are computed on-the-fly by the dataloader from albedo. prev = zero during training (no temporal history). Usage: - python3 pack_photo_sample.py --photo photos/img_001.png \\ + python3 pack_photo_sample.py --photo input/photo1.jpg \\ + --target target_1/photo1_out.png \\ --output dataset/simple/sample_001/ Dependencies: @@ -56,7 +57,7 @@ def pyrdown(img: np.ndarray) -> np.ndarray: # ---- Main packing ---- -def pack_photo_sample(photo_path: str, output_dir: str) -> None: +def pack_photo_sample(photo_path: str, target_path: str, output_dir: str) -> None: os.makedirs(output_dir, exist_ok=True) print(f"[pack_photo_sample] Loading {photo_path} …") @@ -110,9 +111,10 @@ def pack_photo_sample(photo_path: str, output_dir: str) -> None: os.path.join(output_dir, "transp.png") ) - # ---- target — albedo (= photo; no GT styled target) ---- - # Store as RGBA (keep alpha for potential masking by the dataloader). - target_u8 = (np.clip(img_np, 0, 1) * 255.0).astype(np.uint8) + # ---- target — styled ground truth ---- + print(f" Loading target {target_path} …") + target_img = Image.open(target_path).convert("RGBA") + target_u8 = np.asarray(target_img, dtype=np.uint8) Image.fromarray(target_u8, mode="RGBA").save( os.path.join(output_dir, "target.png") ) @@ -129,7 +131,6 @@ def pack_photo_sample(photo_path: str, output_dir: str) -> None: print(" Files: albedo.png normal.png depth.png matid.png " "shadow.png transp.png target.png") print(" Note: normal/depth/matid are zeroed (no geometry data).") - print(" Note: target = albedo (no ground-truth styled target).") def main(): @@ -138,10 +139,12 @@ def main(): ) parser.add_argument("--photo", required=True, help="Input photo file (RGB or RGBA PNG/JPG)") + parser.add_argument("--target", required=True, + help="Styled ground-truth image (PNG/JPG)") parser.add_argument("--output", required=True, help="Output directory for sample files") args = parser.parse_args() - pack_photo_sample(args.photo, args.output) + pack_photo_sample(args.photo, args.target, args.output) if __name__ == "__main__": diff --git a/cnn_v3/training/target_1/photo2_1_out.png b/cnn_v3/training/target_1/photo2_1_out.png deleted file mode 100644 index 5398232..0000000 Binary files a/cnn_v3/training/target_1/photo2_1_out.png and /dev/null differ diff --git a/cnn_v3/training/target_1/photo2_2_out.png b/cnn_v3/training/target_1/photo2_2_out.png deleted file mode 100644 index b0a76bd..0000000 Binary files a/cnn_v3/training/target_1/photo2_2_out.png and /dev/null differ diff --git a/cnn_v3/training/target_1/photo2_out.png b/cnn_v3/training/target_1/photo2_out.png new file mode 100644 index 0000000..5398232 Binary files /dev/null and b/cnn_v3/training/target_1/photo2_out.png differ diff --git a/cnn_v3/training/target_1/photo7_out.png b/cnn_v3/training/target_1/photo7_out.png new file mode 100644 index 0000000..b0a76bd Binary files /dev/null and b/cnn_v3/training/target_1/photo7_out.png differ -- cgit v1.2.3