Spaces:
Runtime error
Runtime error
| set -e | |
| PY=python3 | |
| PARALLEL=$(cat /proc/cpuinfo | grep processor | wc -l) | |
| YOLO=6 | |
| echo "📊 detected ${PARALLEL} cores" | |
| echo "🏛 fetching entities" | |
| ${PY} ./python/get_entities.py | |
| echo "🌏 getting vendor data" | |
| ${PY} ./python/vendor.py --parallel $PARALLEL | |
| echo "☠ getting extra backgrounds from OpenFish" | |
| ${PY} ./python/openfish.py --parallel $PARALLEL | |
| echo "✨ augmenting data" | |
| ${PY} ./python/augment.py --parallel $PARALLEL | |
| echo "🖼 croping augmented data" | |
| ${PY} ./python/crop.py ./data/augmented/images | |
| echo "✂ split dataset into train, val and test groups" | |
| ${PY} ./python/split.py ./data/squares/ --yolo $YOLO | |
| echo "🧠 train model" | |
| sh train.sh | |