mirror of
https://github.com/augustin64/projet-tipe
synced 2025-03-14 06:45:22 +01:00
Add visualisation_kernel.py
This commit is contained in:
parent
635b46bdf3
commit
4a2e3cc3f9
28
src/scripts/visualisation_kernel.py
Normal file
28
src/scripts/visualisation_kernel.py
Normal file
@ -0,0 +1,28 @@
|
|||||||
|
#!/usr/bin/python3
|
||||||
|
#
|
||||||
|
# Use it like that:
|
||||||
|
# ```bash
|
||||||
|
# build/cnn-export print-poids-kernel-cnn -m out.bin > fichier.json
|
||||||
|
# python -c "from src.scripts import visualisation_kernel"
|
||||||
|
# ```
|
||||||
|
import os
|
||||||
|
import json
|
||||||
|
from .visualisation import *
|
||||||
|
|
||||||
|
fichier = "fichier.json" # Fichier d'entrée
|
||||||
|
output_dir = "test-img-vis-kernel" # Dossier de sortie
|
||||||
|
|
||||||
|
with open(fichier, 'r') as f:
|
||||||
|
data = json.load(f)
|
||||||
|
|
||||||
|
IMAGE_WIDTH = 5 # Manière très grossière de modifier le script visualisation.py
|
||||||
|
IMAGE_HEIGHT = 5 # pour utiliser une de ses fonctions sans la réécrire
|
||||||
|
|
||||||
|
os.makedirs(outut_dir, exist_ok=True)
|
||||||
|
for key in data.keys():
|
||||||
|
for i in range(len(data[key])):
|
||||||
|
for j in range(len(data[key][i])):
|
||||||
|
IMAGE_WIDTH = len(data[key][i][j])
|
||||||
|
IMAGE_HEIGHT = len(data[key][i][j][0])
|
||||||
|
png.from_array([[min(int((j2+1)*100),255) for j2 in i2] for i2 in data[key][i][j]], 'L').save(f"{output_dir}/{key}-{i}-{j}.png")
|
||||||
|
|
Loading…
x
Reference in New Issue
Block a user