summaryrefslogtreecommitdiff
path: root/tools/image_importer/imagegen.py
diff options
context:
space:
mode:
authorChunseok Lee <chunseok.lee@samsung.com>2018-09-18 16:53:40 +0900
committerChunseok Lee <chunseok.lee@samsung.com>2018-09-18 16:53:40 +0900
commit91f4ba45449f700a047a4aeea00b1a7c84e94c75 (patch)
treec60eecdba0861c51010fb0519f8a59668d90a6d2 /tools/image_importer/imagegen.py
parent07659ccd9fe7b1cf1547cc6cad78bcf489f0a361 (diff)
downloadnnfw-91f4ba45449f700a047a4aeea00b1a7c84e94c75.tar.gz
nnfw-91f4ba45449f700a047a4aeea00b1a7c84e94c75.tar.bz2
nnfw-91f4ba45449f700a047a4aeea00b1a7c84e94c75.zip
Imported Upstream version 0.2upstream/0.2submit/tizen/20180918.075952
Diffstat (limited to 'tools/image_importer/imagegen.py')
-rwxr-xr-xtools/image_importer/imagegen.py40
1 files changed, 40 insertions, 0 deletions
diff --git a/tools/image_importer/imagegen.py b/tools/image_importer/imagegen.py
new file mode 100755
index 000000000..3c7af0d7f
--- /dev/null
+++ b/tools/image_importer/imagegen.py
@@ -0,0 +1,40 @@
+#!/usr/bin/python
+
+# Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from PIL import Image
+import numpy as np
+
+image_size = {
+ "H": 10,
+ "W": 10,
+ "C": 3 # C is fixed as 3 for R,G,B channels
+}
+
+rgb = np.zeros([image_size['H'], image_size['W'], image_size["C"]], dtype=np.uint8)
+for y in range(image_size["H"]):
+ for x in range(image_size["W"]):
+ for c in range(image_size["C"]):
+ rgb[y][x][c] = 255 #value range = [0~255]
+
+im = Image.fromarray(rgb)
+im.save("image.ppm")
+
+# image can be saved as .jpg or .png
+# im.save("image.jpg")
+# im.save("image.png")
+
+with open("image.bin", "wb") as f:
+ f.write(im.tobytes())