Всё уже давно придумано. Гуглить imagenet keras classification
https://www.pyimagesearch.com/2016/08/10/imagenet-...
Берёшь готовый код и кидаешь в него картинки, а он тебе сразу возвращает результат. Обучать ничего не нужно, потому что всё уже обучено.
Если обязательно нужно самому обучать, то гугли CNN for image classification.