Доработать, чтоб работало. Нейросеть.Tensorflow.Python (Код готов, только подправить)

Есть группа из 20 картинок, поделённых на 2 класса (по типу кошки и собаки). Нужно сделать так, чтоб нейросеть различала их.

Есть шаблонный код, в нем нужно что-то поменять (за этой просьбой сюда и обращаюсь), чтобы при подаче тестовой картинки на выходе выдавало хотя бы больше 80% уверенности в том, что это именно то, что изображено.

Прилагаю:

Neural network - код для нейросети;

Magnetite - первая папка с картинками;

Selen - вторая папка с картинками;

Test picture - картинки для тестирования нейросети.

Если интересно, то на картинках представлены изображения микроскопа для наночастиц магнетита и наночастиц селена:)