Remove resnet from examples/modelzoo/download_model.py
@@ -10,7 +10,7 @@ parser = argparse.ArgumentParser(
description='Download a Caffe reference model')
parser.add_argument('model_type',
choices=('alexnet', 'caffenet', 'googlenet', 'resnet'),
- help='Model type (alexnet, caffenet, googlenet, resnet)')
+ help='Model type (alexnet, caffenet, googlenet)')
args = parser.parse_args()
if args.model_type == 'alexnet':
@@ -24,12 +24,13 @@ elif args.model_type == 'googlenet':
url = 'http://dl.caffe.berkeleyvision.org/bvlc_googlenet.caffemodel'
name = 'bvlc_googlenet.caffemodel'
elif args.model_type == 'resnet':
- url = 'http://research.microsoft.com/en-us/um/people/kahe/resnet/' \
- 'models.zip'
- name = 'models.zip'
+ raise RuntimeError('The resnet model file cannot be downloaded '
+ 'automatically. Please download manually: '
+ 'https://github.com/KaimingHe/deep-residual-networks'
+ '#models')
else:
raise RuntimeError('Invalid model type. Choose from '
- 'alexnet, caffenet, googlenet and resnet.')
+ 'alexnet, caffenet and googlenet.')
print('Downloading model file...')
six.moves.urllib.request.urlretrieve(url, name)
