open rov train

rov_train.py

@@ -412,14 +412,14 @@ def predict_18_19():
 
 
 if __name__ == '__main__':
-    # log_.info('rov model train start...')
-    # train_start = time.time()
-    # train_filename = config_.TRAIN_DATA_FILENAME
-    # X, Y, videos, fea = process_data(filename=train_filename)
-    # log_.info('X_shape = {}, Y_sahpe = {}'.format(X.shape, Y.shape))
-    # train(X, Y, features=fea)
-    # train_end = time.time()
-    # log_.info('rov model train end, execute time = {}ms'.format((train_end - train_start)*1000))
+    log_.info('rov model train start...')
+    train_start = time.time()
+    train_filename = config_.TRAIN_DATA_FILENAME
+    X, Y, videos, fea = process_data(filename=train_filename)
+    log_.info('X_shape = {}, Y_sahpe = {}'.format(X.shape, Y.shape))
+    train(X, Y, features=fea)
+    train_end = time.time()
+    log_.info('rov model train end, execute time = {}ms'.format((train_end - train_start)*1000))
 
     log_.info('rov model predict start...')
     predict_start = time.time()