Написал модель нейронки, которую можно обучить с помощью програмки - трейнера в папкеNeuralNetwork Trainer
В неё нужно написать команду open или train для того чтобы открыть и протестировать модель или начать тренировать новую модель.
Тренеруемая модель будет брать датасет из папок planes_new_train и planes_new_test - Copy. После чего будет сохранять модель в папку models.
Основа: DenceNet201
Мои слои: MaxPool + 2 Dence(relu) + Dence(softmax)
Состоит из 4-х этапов тренировки.
- Обучение нижних, обавленых мной слоёв.
- Обучение с заморозкой 75% слоёв
- Обучение с заморозкой 50% слоёв
- Обучение с заморозкой 25% слоёв
- В будущем планирую обучать всю модель на 5м этапе, но пока случается оверфиттинг с 5го этапа
В папке NeuralNetwork Bot находится исходный код телеграм бота. Он написан с помощью telebot и работает как команда open из трейнера нейронной сети