def run_vgg16_fc6(): vgg16 = fe.CaffeFeatureExtractor( model_path="vgg16_deploy.prototxt", pretrained_path="vgg16.caffemodel", blob="fc6", crop_size=224, mean_values=[103.939, 116.779, 123.68] ) fe.create_dataset(net=vgg16, datalist="train.txt", dbprefix="vgg16_fc6_train") fe.create_dataset(net=vgg16, datalist="test.txt", dbprefix="vgg16_fc6_test")
def run_googlenet(): googlenet = fe.CaffeFeatureExtractor( model_path="googlenet_deploy.prototxt", pretrained_path="googlenet.caffemodel", blob="pool5/7x7_s1", crop_size=224, mean_values=[104.0, 117.0, 123.0] ) fe.create_dataset(net=googlenet, datalist="train.txt", dbprefix="googlenet_train") fe.create_dataset(net=googlenet, datalist="test.txt", dbprefix="googlenet_test")
def run_alexnet(): alexnet = fe.CaffeFeatureExtractor( model_path="alexnet_deploy.prototxt", pretrained_path="alexnet.caffemodel", blob="fc6", crop_size=227, meanfile_path="imagenet_mean.npy" ) fe.create_dataset(net=alexnet, datalist="train.txt", dbprefix="alexnet_train") fe.create_dataset(net=alexnet, datalist="test.txt", dbprefix="alexnet_test")