22 lines
591 B
Python
22 lines
591 B
Python
|
import cv2
|
||
|
|
import tensorflow as tf
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
|
||
|
|
class NeuralNetwork:
|
||
|
|
def __init__(self):
|
||
|
|
self.model = tf.keras.models.load_model("../../files/Neural_networks/model/training_2")
|
||
|
|
|
||
|
|
def Prediction(self, x, y, field):
|
||
|
|
IMG_SIZE = 400
|
||
|
|
img_array = cv2.imread(field.large_image_array_filepath[x][y])
|
||
|
|
new_array = cv2.resize(img_array, (IMG_SIZE, IMG_SIZE))
|
||
|
|
new_array = np.array(new_array) / 255
|
||
|
|
image = new_array.reshape(1, IMG_SIZE, IMG_SIZE, 3)
|
||
|
|
|
||
|
|
prediction = self.model.predict([image])
|
||
|
|
|
||
|
|
return int(prediction[0][0])
|
||
|
|
|
||
|
|
|