Skip to content

xtsxisaxns/alexnet-feature-extraction-finetune

Repository files navigation

AlexNet Feature extraction and finetune

This project uses AlexNet and TensorFlow to build a feature extraction and finetune network.

Files description

feature_extraction_finetune_train.py

extract alexnet feature and add a fc layer,finetune the network.Finally save the weigts

feature_extraction_finetune_predict.py

use the trained network to predict input image

imagenet_inference.py

use alexnet to predict on input image

#Dataset download links Download the dataset.

About

alexnet feature extraction finetune

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •  

Languages