a small fix

divamgupta · Sep 30, 2017 · 15a9b66 · 15a9b66
1 parent 8c0a9d2
commit 15a9b66
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/Models/FCN32.py b/Models/FCN32.py
@@ -58,7 +58,7 @@ def FCN32( n_classes ,  input_height=416, input_width=608 , vgg_level=3):
 	x = Flatten(name='flatten')(x)
 	x = Dense(4096, activation='relu', name='fc1')(x)
 	x = Dense(4096, activation='relu', name='fc2')(x)
-	x = Dense( 1024 , activation='softmax', name='predictions')(x)
+	x = Dense( 1000 , activation='softmax', name='predictions')(x)
 
 	vgg  = Model(  img_input , x  )
 	vgg.load_weights(VGG_Weights_path)

diff --git a/Models/FCN8.py b/Models/FCN8.py
@@ -82,7 +82,7 @@ def FCN8( nClasses ,  input_height=416, input_width=608 , vgg_level=3):
 	x = Flatten(name='flatten')(x)
 	x = Dense(4096, activation='relu', name='fc1')(x)
 	x = Dense(4096, activation='relu', name='fc2')(x)
-	x = Dense( 1024 , activation='softmax', name='predictions')(x)
+	x = Dense( 1000 , activation='softmax', name='predictions')(x)
 
 	vgg  = Model(  img_input , x  )
 	vgg.load_weights(VGG_Weights_path)

diff --git a/Models/VGGSegnet.py b/Models/VGGSegnet.py
@@ -50,7 +50,7 @@ def VGGSegnet( n_classes ,  input_height=416, input_width=608 , vgg_level=3):
 	x = Flatten(name='flatten')(x)
 	x = Dense(4096, activation='relu', name='fc1')(x)
 	x = Dense(4096, activation='relu', name='fc2')(x)
-	x = Dense( 1024 , activation='softmax', name='predictions')(x)
+	x = Dense( 1000 , activation='softmax', name='predictions')(x)
 
 	vgg  = Model(  img_input , x  )
 	vgg.load_weights(VGG_Weights_path)

diff --git a/Models/VGGUnet.py b/Models/VGGUnet.py
@@ -52,7 +52,7 @@ def VGGUnet( n_classes ,  input_height=416, input_width=608 , vgg_level=3):
 	x = Flatten(name='flatten')(x)
 	x = Dense(4096, activation='relu', name='fc1')(x)
 	x = Dense(4096, activation='relu', name='fc2')(x)
-	x = Dense( 1024 , activation='softmax', name='predictions')(x)
+	x = Dense( 1000 , activation='softmax', name='predictions')(x)
 
 	vgg  = Model(  img_input , x  )
 	vgg.load_weights(VGG_Weights_path)