TransWikia.com

How to fix debuged CNN model in Keras

Data Science Asked by DispooL on December 23, 2020

I’ve trained a CNN model based on landscape and pool images. The main purpose is to make model classify if the image contains a pool or not. The accuracy on the test set was about 94% but when I used my own image to test, sometimes model did wrong predictions then I implemented GradCam and realized that the model looks only at the edges of the pool when I would classify the pool by the water. How can I fix my model so it could make better predictions and look not only at the edges?

enter image description here

One Answer

I solved the problem, I added an additional conv layer with 64 feature maps and increased size of dense layer from 512 to 1024 also I trained the model with the increased number of epochs(from 50 to 150). The result is impressive! Everything as I wanted.

enter image description here

Answered by DispooL on December 23, 2020

Add your own answers!

Ask a Question

Get help from others!

© 2024 TransWikia.com. All rights reserved. Sites we Love: PCI Database, UKBizDB, Menu Kuliner, Sharing RPP