mirror of
https://github.com/prise6/smart-iss-posts
synced 2024-04-27 03:21:51 +02:00
22 lines
765 B
Python
22 lines
765 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
||
|
from iss.tools import Config
|
||
|
from iss.tools import Tools
|
||
|
from iss.models import SimpleConvAutoEncoder
|
||
|
from iss.clustering import ClassicalClustering
|
||
|
from dotenv import find_dotenv, load_dotenv
|
||
|
|
||
|
## Config
|
||
|
load_dotenv(find_dotenv())
|
||
|
cfg = Config(project_dir = os.getenv("PROJECT_DIR"), mode = os.getenv("MODE"))
|
||
|
|
||
|
## charger le modèle
|
||
|
model_type = 'simple_conv'
|
||
|
cfg.get('models')[model_type]['model_name'] = 'model_colab'
|
||
|
model = SimpleConvAutoEncoder(cfg.get('models')[model_type])
|
||
|
|
||
|
## Générateur d'image
|
||
|
filenames = Tools.list_directory_filenames('data/processed/models/autoencoder/train/k/')
|
||
|
generator_imgs = Tools.generator_np_picture_from_filenames(filenames, target_size = (27, 48), batch = 496, nb_batch = 2)
|
||
|
|
||
|
## Générer des images
|