s462072/Trashmaster
s462072/Trashmaster
5
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

hotfix/netural_network #29

Merged
s464843 merged 3 commits from hotfix/netural_network into master 2022-05-26 21:48:29 +02:00
Conversation 0 Commits 3 Files Changed 4 +2 -2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 49db999cc9 - Show all commits

4
NeuralNetwork/prediction.py
View File

@ -5,9 +5,9 @@ from PIL import Image
from NeuralNetwork import NeuralNetwork from NeuralNetwork import NeuralNetwork
def getPrediction(img_path): def getPrediction(img_path):
# Inicjacja sieci neuronowej # Inicjacja sieci neuronowej
neural_net = NeuralNetwork() neural_net = NeuralNetwork.NeuralNetwork()
PATH = './trained_nn.pth' PATH = './trained_nn.pth'
img = Image.open(img_path) img = Image.open(img_path)
transform_tensor = transforms.ToTensor()(img).unsqueeze_(0) transform_tensor = transforms.ToTensor()(img).unsqueeze_(0)