train_model.py 文件源码

def train():
    # stack_optical_flow(file_directory, data_update=False)
    with open(pickle_directory + 'class_index_dict.pickle', 'rb') as fr:
        class_index_dict = pickle.load(fr)
    # num_of_classes = int(len(class_index_dict) / 2)
    # seed = [random.random() for i in range(num_of_classes)]

    print('Training temporal model.')
    train_temporal_model(class_index_dict)
    gc.collect()

    # release memory
    # ------------------------
    K.clear_session()
    # sess = tf.Session()
    # K.set_session(sess)
    # ------------------------

    # print('Training spatial model.')
    # train_spatial_model(class_index_dict)
    # gc.collect()

    print('ok.')