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

repair getPrediction

Browse Source
This commit is contained in:
happy531 2022-05-26 20:55:34 +02:00
parent 134e45de18
commit 49db999cc9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
NeuralNetwork/prediction.py
View File

@ -7,7 +7,7 @@ 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)