BubbleProject/data_gen.sh at main · nmazda/BubbleProject · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#!/bin/bash

RUNS_DIR="/home/iec/Documents/bubble_project/BubbleProject/bubble_detector/runs"

# TODO: Need to actually use the arguments
SRC_IMGS=$1
SPLIT_OUT=$2
SPLT_AMNT=$3
CHPT=$4

# Move to dir: bubble_detector
cd /home/iec/Documents/bubble_project/BubbleProject/bubble_detector

# Creates BW versions of Real images
python detect_BW.py ./models/bubble_swin-b/config.py $CHPT $SRC_IMGS
# Gets most recent detect folder
MOST_RECENT_DETECT=$(ls -td "$RUNS_DIR"/detect* 2>/dev/null | head -n 1)

# Move to dir: bubble_augmentation
cd /home/iec/Documents/bubble_project/BubbleProject/bubble_augmentation

if [ ! -d "$SPLIT_OUT/real_split" ]; then
    echo "$SPLIT_OUT/real_split does not exist. Creating it..."
    mkdir -p "$SPLIT_OUT/real_split"
fi

# Runs MirrorAndSplit with both mirroring and non mirroring for both real and bw
# Real Mirror then Non-Mirror
python ./mirrorAndSplit.py -m -s $SPLT_AMNT $SRC_IMGS $SPLIT_OUT/real_split
python ./mirrorAndSplit.py -s $SPLT_AMNT $SRC_IMGS $SPLIT_OUT/real_split

if [ ! -d "$SPLIT_OUT/bw_split" ]; then
    echo "$SPLIT_OUT/bw_split does not exist. Creating it..."
    mkdir -p "$SPLIT_OUT/bw_split"
fi

# BW Mirror then Non-Mirror
python ./mirrorAndSplit.py -m -s $SPLT_AMNT $MOST_RECENT_DETECT $SPLIT_OUT/bw_split
python ./mirrorAndSplit.py -s $SPLT_AMNT $MOST_RECENT_DETECT $SPLIT_OUT/bw_split