Chooses 20 samples from the input directory and run Faster R-CNN

run_frcnn.sh

#!/bin/bash

if [ $# -lt 2 ] 
then
    echo "usage: $0 [input_dir] [output_dir]"
    echo "chooses 20 video files randomly from the input directory and run Faster R-CNN"
    exit 1
fi

input_dir="$1"
output_dir="$2"

files=$(find "$input_dir" -type f -name "*.avi" -o -name "*.mp4" -o -name "*.MP4" | shuf | head -20)

input_dir_parent=$(dirname "$input_dir")

while read line
do
    input_file="$line"
    output_file=$(echo "$input_file" | sed "s|$input_dir_parent|$output_dir|")

    echo "Processing $input_file"
    echo "Saving to $output_file"

    mkdir -p $(dirname "$output_file")
    python3 /detectron2_repo/demo/demo.py --config-file /detectron2_repo/configs/COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml --video-input "$input_file" --confidence-threshold 0.6 --output "$output_file" --opts MODEL.WEIGHTS detectron2://COCO-Detection/faster_rcnn_R_101_FPN_3x/137851257/model_final_f6e8b1.pkl
done <<< "$files"