diff --git a/scripts/data.py b/scripts/data.py index 0873d1d..5083d40 100644 --- a/scripts/data.py +++ b/scripts/data.py @@ -96,7 +96,7 @@ with open(filePath, 'r') as file: if line[0:4] == 'nc: ' : parts = line.strip().split(': ') jsonObj['config']['classes'] = int(parts[1]) - else if line[0:7] == 'names: ' : + elif line[0:7] == 'names: ' : classesLine = line.strip().replace('names: ', '').replace('[', '').replace(']', '') classes = classesLine.split(',') i = 0