Skip to content

arpankapoor/pycuda-vgg16

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

A VGG16 inference module comparison between

  • tensorflow
  • numpy
  • pycuda

execute

  • required dependencies were installed with miniforge using conda install pillow pycuda tensorflow-gpu
  • VGG16 weights pretrained on ImageNet were downloaded from here and placed in the same directory as the code in this repository
$ ./vgg.py tf /path/to/image_to_classify      # run inference using tensorflow
$ ./vgg.py numpy /path/to/image_to_classify   # run inference using numpy implementation
$ ./vgg.py pycuda /path/to/image_to_classify  # run inference using pycuda implementation

About

vgg16 inference implementation using tensorflow, numpy and pycuda

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages