File size: 671 Bytes
e919aa3
bbaa3e0
 
e919aa3
fac6e9b
eb42660
bbaa3e0
fac6e9b
e919aa3
 
 
fac6e9b
 
 
e919aa3
c7f1171
e919aa3
 
b1d65c2
eb42660
bbaa3e0
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#!/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