Open GreNait opened 4 years ago
I generated different models (6 up to now) which are all similar but different in neurons and convolutional layers. The idea is to get a better overview of the capabilities of a model.
It appears, that something is completly off. I get from the beginning "nan" in the loss function.
I made a heavy mistake, with the categorie in the default parameters. I have 3 figures and only provided 2 labels. Hterefore, the nan appeared. I changed that, and the model now starts training with a loss.
All three models (excluding non cnns) are able to detect somehow the figures. But they are not perfect. Antman is discovered as batman or vice versa. My guess is, that they all tend to overfitting therefore have problems with the recognition. I will try it again, this time with a dropout rate.
Normally, the first 4 should be a 0 (antman) -> cnn_small
Treined the big model with the Dropout, and the detection workded:
I found finally a way to get the steps per epoch right. The imageDtaGenerator has a way to provide that. Just use: "imageDataGenerator2.samples. Thats it. TOTAL_TRAIN can than easily calculated by = self.train_data_generator.samples/self.train_data_generator.batch_size
Generate different modeld to evaluate their performence with different parameters: