From 26cf6e733d8734ced2096d9d3ceb27ccd75b72be Mon Sep 17 00:00:00 2001 From: matixezor Date: Sun, 30 May 2021 18:01:07 +0200 Subject: [PATCH] test --- src/machine_learning/neural_network/changingNameOfFiles.py | 2 +- src/machine_learning/neural_network/net.py | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/machine_learning/neural_network/changingNameOfFiles.py b/src/machine_learning/neural_network/changingNameOfFiles.py index bc861ab..ffe7865 100644 --- a/src/machine_learning/neural_network/changingNameOfFiles.py +++ b/src/machine_learning/neural_network/changingNameOfFiles.py @@ -53,7 +53,7 @@ def resize_images(): if im.mode != 'RGB': im = im.convert ('RGB') #f, e = os.path.splitext (item) - imResize = im.resize((28, 28), Image.ANTIALIAS) + imResize = im.resize((256, 256), Image.ANTIALIAS) imResize.save(f'{DIR}/{item}', quality=100) if __name__ == '__main__': diff --git a/src/machine_learning/neural_network/net.py b/src/machine_learning/neural_network/net.py index db51007..1fe18c8 100644 --- a/src/machine_learning/neural_network/net.py +++ b/src/machine_learning/neural_network/net.py @@ -6,9 +6,11 @@ class NeuralNetwork(nn.Module): super(NeuralNetwork, self).__init__() self.flatten = nn.Flatten() self.linear_relu_stack = nn.Sequential( - nn.Linear(28*28*3, 512), + nn.Linear(256*256*3, 2048), nn.ReLU(), - nn.Linear(512, 512), + nn.Linear(2048, 1024), + nn.ReLU(), + nn.Linear(1024, 512), nn.ReLU(), nn.Linear(512, 10), nn.ReLU()