Only read pcaps
@@ -40,6 +40,8 @@ def read_data(data_dir, duration=None, labels=None):
for dirpath, dirnames, filenames in os.walk(data_dir):
for file in filenames:
+ _, ext = os.path.splitext(file)
+ if ext == '.pcap':
files.append(os.path.join(dirpath,file))
# Go through all the files in the directory
for filename in files:
