diff --git a/src/scripts/visualisation_kernel.py b/src/scripts/visualisation_kernel.py new file mode 100644 index 0000000..d77a005 --- /dev/null +++ b/src/scripts/visualisation_kernel.py @@ -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") +