a small fix

Author: Divam Gupta

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)

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)

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)

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)