Commit a69d5335 authored by Anthony Larcher's avatar Anthony Larcher
Browse files

sad

parent 4212d4da
......@@ -17,8 +17,8 @@ class SAD_Dataset(Dataset):
def __init__(self, mdtm_file, features_server, batch_size=512, duration=3.2, step=0.8, uem_file=None, shuffle=False):
self.batch_size = batch_size
self.duration = duration * 100
self.step = step * 100
self.duration = int(duration * 100)
self.step = int(step * 100)
self.features_server = features_server
train_list = {}
......@@ -30,8 +30,7 @@ class SAD_Dataset(Dataset):
train_list[show].append({
"start": int(float(start) * 100),
"stop": int((float(start) + float(dur)) * 100)})
#
#GERER LE CAS OU UEM=NONE, ON PREND LES FICHIERS COMPLETS
uem_list = {}
if uem_file is not None:
with open(uem_file, 'r') as f:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment