Scripts for Inference using your pre-trained models from Google Drive

he-y / soft-filter-pruning

Soft Filter Pruning for Accelerating Deep Convolutional Neural Networks

https://arxiv.org/abs/1808.06866

374 stars 74 forks source link

Scripts for Inference using your pre-trained models from Google Drive #1

Closed dhingratul closed 5 years ago

dhingratul commented 5 years ago

Is the inference only scripts also available?

he-y commented 5 years ago

Please refer to scripts/inference_resnet.sh and the updated README.md

dhingratul commented 5 years ago

@he-y When you say "/path/to/ImageNet2012" , what is the structure of the folder, is the ILSVRC2012_img_train or val ?

he-y commented 5 years ago

We follow the Facebook process of ImageNet. Two subfolders ("train" and "val") are included in the "/path/to/ImageNet2012". The correspding code is here.