Open JosephMartinelli opened 3 years ago
Maybe I've found a solution but i really don't know if it's the optimal way to do it since I've hard coded it in extractor.py but there is it:
In extractor.py there is a function called:
def extract(self, image_path):
img = image.load_img(image_path)
return self.extract_image(img)
I have modified it in this way:
def extract(self, image_path):
img = image.load_img(image_path,target_size=(224,224,3))
return self.extract_image(img)
So I have forced the reshaping of the image loaded. The train.py seems running fine. I'll share any issues I'll find along the training
Hello! I'm having this error while executing train.py. Although the target shape is correct i can't wrap my head around the expected shape.
I'll leave the stack trace: traceback.txt