mirror of
https://github.com/augustin64/projet-tipe
synced 2025-03-13 22:35:21 +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