Skip to content

Instantly share code, notes, and snippets.

@Praveen76

Praveen76/Define labels & write them in a file.py

Created July 29, 2020 05:29
Show Gist options
  • Save Praveen76/1590d1c334ce734831311b53f2f23ea4 to your computer and use it in GitHub Desktop.
Save Praveen76/1590d1c334ce734831311b53f2f23ea4 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Define labels & write them in a file.py
#Define labels & write them in a file
classes = ['mask','noMask']
with open('../maskDetectorClasses.csv', 'w') as f:
for i, class_name in enumerate(classes):
f.write(f'{class_name},{i}\n')
if not os.path.exists('snapshots'):
os.mkdir('snapshots')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment