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