25 lines
680 B
Bash
Executable File
25 lines
680 B
Bash
Executable File
#!/bin/bash
|
|
|
|
all_label_file="data/roi/train/labels.txt"
|
|
npos=$(grep '1$' $all_label_file | wc -l)
|
|
nneg=$(grep '0$' $all_label_file | wc -l)
|
|
|
|
echo "npos: $npos"
|
|
echo "nneg: $nneg"
|
|
|
|
min=$npos
|
|
if [ $nneg -lt $min ]; then
|
|
min=$nneg
|
|
fi
|
|
|
|
echo "min: $min"
|
|
cat $all_label_file | grep '1$' | shuf | head -n $min > data/roi/train/pos.txt
|
|
cat $all_label_file | grep '0$' | shuf | head -n $min > data/roi/train/neg.txt
|
|
|
|
cat data/roi/train/pos.txt data/roi/train/neg.txt > data/roi/train/mixed.txt
|
|
|
|
# these are the 'original' prints with different bg pattern offset
|
|
grep -e "7792[1-7].jpg" $all_label_file >> data/roi/train/mixed.txt
|
|
|
|
./roi-train.py --labels-file data/roi/train/mixed.txt
|