Skip to content

Minimalist implementation of ImageNet prediction using ResNet50 with Keras and Docker

Notifications You must be signed in to change notification settings

ccivit/simple_imagenet

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Simple ImageNet classifier

Minimalist implementation of ImageNet prediction using ResNet101 with Keras and Docker

git clone https://github.com/ccivit/simple_imagenet
cd simple_imagenet
sudo nvidia-docker build -t imagenet .

This will build the docker. In order to get the imagenet classification for a specific image, run the following command:

nvidia-docker run -it -v /volume/to/map:/keras imagenet /image/path

About

Minimalist implementation of ImageNet prediction using ResNet50 with Keras and Docker

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published