Skip to content

YYlin/Collection-of-Image-Recognition-Models-in-Tensorflow

Repository files navigation

Image recognition model collection

Collection of classic image recognition models, e.g.ResNet, Alexnet, VGG19, inception_V4 in Tensorflow.

The format of dataset

Firstly, Use unzip data.zip to unzipped it, and then images belonging to different categories are placed in different folders.

Training networks

models paper commands introduction
Alexnet Alexnet python main.py --type_of_model Alexnet Alexnet
VGG19 VGG19 python main.py --type_of_model VGG19 VGG19
Inception_V4 Inception_V4 python main.py --type_of_model inception_V4 Inception_V4
ResNet ResNet python main.py --type_of_model ResNet --resnet_type 50 ResNet

The Accuracy of training and validing images

An explanation

It's easy to use the above recognition models with kears, but by reading papers and analyzing source code, which could help in improving programming ability. If you want to know basic something about these models, you can visit my CSDN,

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages