From 565dd0f695b46b49b9ec32874339b979b1c9d341 Mon Sep 17 00:00:00 2001 From: HombreLaser Date: Fri, 20 Oct 2023 21:09:08 -0600 Subject: Fix weights generation --- main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'main.py') diff --git a/main.py b/main.py index 9ca0a13..a44fbcb 100644 --- a/main.py +++ b/main.py @@ -1,12 +1,12 @@ -from PIL import Image -from dataset_utils import DatasetUtils +from dataset import Dataset from neural_network import NeuralNetwork LEARNING_RATE = .5 INPUT_RESOLUTION = 278 + def main(): - data = DatasetUtils() + data = Dataset() image = data.get_image('А/5a2f3c19c27bb.png') neural_network = NeuralNetwork(LEARNING_RATE, INPUT_RESOLUTION) -- cgit v1.2.3