summaryrefslogtreecommitdiff
path: root/cnn_v3/training
diff options
context:
space:
mode:
Diffstat (limited to 'cnn_v3/training')
-rw-r--r--cnn_v3/training/dataset/simple/sample_006/albedo.pngbin2363392 -> 2059642 bytes
-rw-r--r--cnn_v3/training/dataset/simple/sample_006/depth.pngbin2988 -> 390 bytes
-rw-r--r--cnn_v3/training/dataset/simple/sample_006/matid.pngbin1534 -> 390 bytes
-rw-r--r--cnn_v3/training/dataset/simple/sample_006/normal.pngbin7774 -> 451 bytes
-rw-r--r--cnn_v3/training/dataset/simple/sample_006/shadow.pngbin4399 -> 713 bytes
-rw-r--r--cnn_v3/training/dataset/simple/sample_006/target.pngbin2798038 -> 1617556 bytes
-rw-r--r--cnn_v3/training/dataset/simple/sample_006/transp.pngbin1534 -> 390 bytes
-rw-r--r--cnn_v3/training/export_cnn_v3_weights.py9
8 files changed, 9 insertions, 0 deletions
diff --git a/cnn_v3/training/dataset/simple/sample_006/albedo.png b/cnn_v3/training/dataset/simple/sample_006/albedo.png
index 6fd43d4..0de8932 100644
--- a/cnn_v3/training/dataset/simple/sample_006/albedo.png
+++ b/cnn_v3/training/dataset/simple/sample_006/albedo.png
Binary files differ
diff --git a/cnn_v3/training/dataset/simple/sample_006/depth.png b/cnn_v3/training/dataset/simple/sample_006/depth.png
index 4ccb0bc..9379478 100644
--- a/cnn_v3/training/dataset/simple/sample_006/depth.png
+++ b/cnn_v3/training/dataset/simple/sample_006/depth.png
Binary files differ
diff --git a/cnn_v3/training/dataset/simple/sample_006/matid.png b/cnn_v3/training/dataset/simple/sample_006/matid.png
index 32e1b1c..9379478 100644
--- a/cnn_v3/training/dataset/simple/sample_006/matid.png
+++ b/cnn_v3/training/dataset/simple/sample_006/matid.png
Binary files differ
diff --git a/cnn_v3/training/dataset/simple/sample_006/normal.png b/cnn_v3/training/dataset/simple/sample_006/normal.png
index e2623bd..5888a55 100644
--- a/cnn_v3/training/dataset/simple/sample_006/normal.png
+++ b/cnn_v3/training/dataset/simple/sample_006/normal.png
Binary files differ
diff --git a/cnn_v3/training/dataset/simple/sample_006/shadow.png b/cnn_v3/training/dataset/simple/sample_006/shadow.png
index 880ccf9..c92a2e9 100644
--- a/cnn_v3/training/dataset/simple/sample_006/shadow.png
+++ b/cnn_v3/training/dataset/simple/sample_006/shadow.png
Binary files differ
diff --git a/cnn_v3/training/dataset/simple/sample_006/target.png b/cnn_v3/training/dataset/simple/sample_006/target.png
index 495ed83..6107560 100644
--- a/cnn_v3/training/dataset/simple/sample_006/target.png
+++ b/cnn_v3/training/dataset/simple/sample_006/target.png
Binary files differ
diff --git a/cnn_v3/training/dataset/simple/sample_006/transp.png b/cnn_v3/training/dataset/simple/sample_006/transp.png
index 32e1b1c..32ff022 100644
--- a/cnn_v3/training/dataset/simple/sample_006/transp.png
+++ b/cnn_v3/training/dataset/simple/sample_006/transp.png
Binary files differ
diff --git a/cnn_v3/training/export_cnn_v3_weights.py b/cnn_v3/training/export_cnn_v3_weights.py
index 6d99af9..99f3a81 100644
--- a/cnn_v3/training/export_cnn_v3_weights.py
+++ b/cnn_v3/training/export_cnn_v3_weights.py
@@ -1,4 +1,13 @@
#!/usr/bin/env python3
+# /// script
+# requires-python = ">=3.11"
+# dependencies = [
+# "numpy",
+# "opencv-python",
+# "pillow",
+# "torch",
+# ]
+# ///
"""Export trained CNN v3 weights → binary files for C++ runtime.
Outputs