Skip to content

TechNexion-customization/tn-ml

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

classification

Repository for emDNN compressed classification models.

CIFAR-10 classification

  • Classification using emDNN compressed ResNet-18 can be done using:
    python cifar_classifier.py -p models/resnet18/resnet18_emdnn.prototxt -m models/resnet18/resnet18_emdnn.caffemodel
    
  • To evaluate pretrained model use normalized CIFAR-10 images, using mean = (0.4914, 0.4822, 0.4465) and scaled down by 255.0 and save it to a npz file to be read by eval_cifar_caffe_model.py or else download already normalized CIFAR-10 from here.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages