.gitignore

# Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]
*$py.class

.DS_Store
pretrained_weights
output

venv/
pretrained_weights/
.idea/