Update inception_v4.py

kentsommer · Feb 3, 2017 · f4c8302 · f4c8302
1 parent 1c143c1
commit f4c8302
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/inception_v4.py b/inception_v4.py
@@ -16,8 +16,8 @@
 # Implements the Inception Network v4 (http://arxiv.org/pdf/1602.07261v1.pdf) in Keras. #
 #########################################################################################
 
-TH_WEIGHTS_PATH = 'https://github.com/kentsommer/keras-inceptionV4/releases/download/1.0/inception-v4_weights_th_dim_ordering_th_kernels.h5'
-TF_WEIGHTS_PATH = 'https://github.com/kentsommer/keras-inceptionV4/releases/download/1.0/inception-v4_weights_tf_dim_ordering_tf_kernels.h5'
+TH_WEIGHTS_PATH = 'https://github.com/kentsommer/keras-inceptionV4/releases/download/2.0/inception-v4_weights_th_dim_ordering_th_kernels.h5'
+TF_WEIGHTS_PATH = 'https://github.com/kentsommer/keras-inceptionV4/releases/download/2.0/inception-v4_weights_tf_dim_ordering_tf_kernels.h5'
 
 def conv2d_bn(x, nb_filter, nb_row, nb_col,
               border_mode='same', subsample=(1, 1), bias=False):
@@ -273,4 +273,4 @@ def inception_v4(num_classes, dropout_keep_prob, weights):
     return model
 
 def create_model(num_classes=1001, dropout_keep_prob=0.8, weights=None):
-	return inception_v4(num_classes, dropout_keep_prob, weights)
+	return inception_v4(num_classes, dropout_keep_prob, weights)