fix

Author: GreenWizard2015

Core/Utils.py

@@ -272,6 +272,7 @@ def extractSessions(dataset, TDelta):
   res = []
   T = 0
   prevSession = 0
+  N = len(dataset['time'])
   for i, t in enumerate(dataset['time']):
     if TDelta < (t - T):
       if 1 < (i - prevSession):
@@ -281,10 +282,13 @@ def extractSessions(dataset, TDelta):
     T = t
     continue
   # if last session is not empty, then append it
-  if prevSession < len(dataset['time']):
-    res.append((prevSession, len(dataset['time'])))
+  if prevSession < N:
+    res.append((prevSession, N))
     pass
 
+  # remove sessions with less than 2 samples
+  res = [x for x in res if 1 < (x[1] - x[0])]
+
   # check that end of one session is equal or less than start of the next
   for i in range(1, len(res)):
     assert res[i-1][1] <= res[i][0]

scripts/preprocess-remote.py

@@ -194,7 +194,7 @@ def processFolder(folder, timeDelta, testRatio, framesPerChunk, testPadding, ski
 
   if (0 == len(training)) or (0 == len(testing)):
     print('No training or testing sets found!')
-    return 0, 0, True
+    return 0, 0, True, None
 
   def saveSubset(filename, idx):
     print('%s: %d frames' % (filename, len(idx)))