예제 #1
0
                                      classes=classes,
                                      weights=None,
                                      dropout=dropout,
                                      fine=1,
                                      retrain=False,
                                      pre_file=pre_file,
                                      old_epochs=old_epochs,
                                      cross_index=cross_index,
                                      input_shape=spatial_a.output_shape,
                                      input_tensor=spatial_a.output)

spatial = models.InceptionSpatial(n_neurons=n_neurons,
                                  seq_len=seq_len,
                                  classes=classes,
                                  weights=None,
                                  dropout=dropout,
                                  fine=False,
                                  retrain=False,
                                  pre_file=pre_file,
                                  old_epochs=old_epochs,
                                  cross_index=cross_index)

spatial_a.summary()
spatial_b.summary()
spatial.summary()

# print(glob.glob('weights/inception_spatial2fc_{}_{}e_cr{}.h5'.format(n_neurons,pre_train[0],cross_index)[-1]))
spatial.load_weights('weights/sinception_spatial2fc_256-62-0.8554.hdf5')
print("Load")

spatial_a.layers[0].set_weights(spatial.layers[0].get_weights())
예제 #2
0
n_neurons = args.neural
dropout = args.dropout
pre_file = 'inception_spatial2fc_{}'.format(n_neurons)

if train & (not retrain):
    weights = 'imagenet'
else:
    weights = None

if args.fine == 1:
    fine = True
else:
    fine = False

result_model = models.InceptionSpatial(
                    n_neurons=n_neurons, seq_len=seq_len, classes=classes, 
                    weights=weights, dropout=dropout, fine=fine, retrain=retrain,
                    pre_file=pre_file,old_epochs=old_epochs,cross_index=cross_index)

if (args.summary == 1):
    result_model.summary()
    sys.exit()

lr = args.lr 
decay = args.decay

losses = {
	"loss1": "categorical_crossentropy",
	"loss2": "categorical_crossentropy",
    "loss3": "categorical_crossentropy"
}
lossWeights = {"loss1": 1.0, "loss2": 1.0, "loss3": 1.0}