Skip to content

shashwat14/PreTrainedVGG16

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

PreTrainedVGG16

This a pre trained VGG16 model that is meant for usage in predicting, finetuning or entirely new projects. This VGG16 network also provides extensive visualization during training for the purpose of debugging (and) or hyperparamter seach. This project also is literally plug and play for the first two cases mentioned below. Example usage scenarios:

  1. Classify image in 1 among 1000 categories of ImageNet - Not that useful
  2. Finetune VGG16 for classifying images from datasets apart from ImageNet - Useful
  3. Using the the convolutoinal codes (25088 dimension) vector and appending your custom layers - Very useful

Installation

I have borrowed the pre-trained weights from the following link - https://www.cs.toronto.edu/~frossard/post/vgg16/

  1. Git clone this project. Let's call the directory $root
  2. Download the weights file (vgg16_weights.npz) and place in $root

Classifying images in 1 among 1000 categories form ImageNet

To be finished

About

A pre-trained VGG for fine-tuning and usage in other networks

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages