summaryrefslogtreecommitdiff
path: root/doc/HOWTO.md
diff options
context:
space:
mode:
Diffstat (limited to 'doc/HOWTO.md')
-rw-r--r--doc/HOWTO.md9
1 files changed, 9 insertions, 0 deletions
diff --git a/doc/HOWTO.md b/doc/HOWTO.md
index 2290aa8..1e8b58b 100644
--- a/doc/HOWTO.md
+++ b/doc/HOWTO.md
@@ -145,6 +145,9 @@ Enhanced CNN with parametric static features (7D input: RGBD + UV + sin encoding
# Custom architecture
./scripts/train_cnn_v2_full.sh --kernel-sizes 3,5,3 --num-layers 3 --mip-level 1
+# Grayscale loss (compute loss on luminance instead of RGBA)
+./scripts/train_cnn_v2_full.sh --grayscale-loss
+
# Custom directories
./scripts/train_cnn_v2_full.sh --input training/input --target training/target_2
@@ -188,6 +191,12 @@ Enhanced CNN with parametric static features (7D input: RGBD + UV + sin encoding
--input training/input/ --target training/target_2/ \
--mip-level 1 \
--epochs 100 --batch-size 16
+
+# Grayscale loss (compute loss on luminance Y = 0.299*R + 0.587*G + 0.114*B)
+./training/train_cnn_v2.py \
+ --input training/input/ --target training/target_2/ \
+ --grayscale-loss \
+ --epochs 100 --batch-size 16
```
**Export Binary Weights:**