From 0db3656406f5da90ad87f3a3be7144a07539509d Mon Sep 17 00:00:00 2001 From: Matt Harvey Date: Mon, 20 Nov 2017 18:58:57 -0800 Subject: [PATCH] Settings --- extract_features.py | 4 ++-- train.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/extract_features.py b/extract_features.py index 64ee4f1..dbc9ee0 100644 --- a/extract_features.py +++ b/extract_features.py @@ -19,7 +19,7 @@ # Set defaults. seq_length = 40 -class_limit = 2 # Number of classes to extract. Can be 1-101 or None for all. +class_limit = None # Number of classes to extract. Can be 1-101 or None for all. # Get the dataset. data = DataSet(seq_length=seq_length, class_limit=class_limit) @@ -36,7 +36,7 @@ '-features') # numpy will auto-append .npy # Check if we already have it. - if os.path.isfile(path): + if os.path.isfile(path + '.npy'): pbar.update(1) continue diff --git a/train.py b/train.py index f931b3f..bceb2ed 100644 --- a/train.py +++ b/train.py @@ -86,7 +86,7 @@ def main(): # model can be one of lstm, lrcn, mlp, conv_3d, c3d model = 'lstm' saved_model = None # None or weights file - class_limit = 10 # int, can be 1-101 or None + class_limit = None # int, can be 1-101 or None seq_length = 40 load_to_memory = False # pre-load the sequences into memory batch_size = 32