run.sh

#!/bin/sh
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