examples.sh

#!/bin/bash
# Example commands for ROI-VAE compression
# Covers: Car, Building, Person, Boat

# Make sure you're in the roi-vae directory
cd "$(dirname "$0")"

# Create results directory if it doesn't exist
mkdir -p results

echo "ROI-VAE Compression Examples"
echo "============================="
echo

# ---------------------------------------------------------
# 1. CAR
# ---------------------------------------------------------
echo "1. CAR (YOLO)"
echo "-------------"

# 1a. Only Image
echo "a) Compressing (Image Only)..."
python roi_compressor.py \
    --input data/images/car/0016cf15fa4d4e16.jpg \
    --output results/car_compressed.jpg \
    --checkpoint checkpoints/tic_lambda_0.0483.pth.tar \
    --sigma 0.3 \
    --seg-method yolo \
    --seg-classes car

# 1b. With Comparison
echo "b) Compressing (With Comparison)..."
python roi_compressor.py \
    --input data/images/car/0016cf15fa4d4e16.jpg \
    --output results/car_comparison.jpg \
    --checkpoint checkpoints/tic_lambda_0.0483.pth.tar \
    --sigma 0.3 \
    --seg-method yolo \
    --seg-classes car \
    --highlight

echo
echo "---------------------------------------------------------"
echo

# ---------------------------------------------------------
# 2. BUILDING
# ---------------------------------------------------------
echo "2. BUILDING (SegFormer)"
echo "-----------------------"

# 2a. Only Image
echo "a) Compressing (Image Only)..."
python roi_compressor.py \
    --input data/images/building/000571767ec7a593.jpg \
    --output results/building_compressed.jpg \
    --checkpoint checkpoints/tic_lambda_0.0483.pth.tar \
    --sigma 0.3 \
    --seg-classes building

# 2b. With Comparison
echo "b) Compressing (With Comparison)..."
python roi_compressor.py \
    --input data/images/building/000571767ec7a593.jpg \
    --output results/building_comparison.jpg \
    --checkpoint checkpoints/tic_lambda_0.0483.pth.tar \
    --sigma 0.3 \
    --seg-classes building \
    --highlight

echo
echo "---------------------------------------------------------"
echo

# ---------------------------------------------------------
# 3. PERSON
# ---------------------------------------------------------
echo "3. PERSON (YOLO)"
echo "----------------"

# 3a. Only Image
echo "a) Compressing (Image Only)..."
python roi_compressor.py \
    --input data/images/person/kodim04.png \
    --output results/person_compressed.jpg \
    --checkpoint checkpoints/tic_lambda_0.0483.pth.tar \
    --sigma 0.3 \
    --seg-method yolo \
    --seg-classes person

# 3b. With Comparison
echo "b) Compressing (With Comparison)..."
python roi_compressor.py \
    --input data/images/person/kodim04.png \
    --output results/person_comparison.jpg \
    --checkpoint checkpoints/tic_lambda_0.0483.pth.tar \
    --sigma 0.3 \
    --seg-method yolo \
    --seg-classes person \
    --highlight

echo
echo "---------------------------------------------------------"
echo

# ---------------------------------------------------------
# 4. BOAT
# ---------------------------------------------------------
echo "4. BOAT (YOLO)"
echo "--------------"

# 4a. Only Image
echo "a) Compressing (Image Only)..."
python roi_compressor.py \
    --input data/images/boat/kodim06.png \
    --output results/boat_compressed.jpg \
    --checkpoint checkpoints/tic_lambda_0.0483.pth.tar \
    --sigma 0.3 \
    --seg-method yolo \
    --seg-classes boat

# 4b. With Comparison
echo "b) Compressing (With Comparison)..."
python roi_compressor.py \
    --input data/images/boat/kodim06.png \
    --output results/boat_comparison.jpg \
    --checkpoint checkpoints/tic_lambda_0.0483.pth.tar \
    --sigma 0.3 \
    --seg-method yolo \
    --seg-classes boat \
    --highlight

echo
echo "---------------------------------------------------------"
echo

# ---------------------------------------------------------
# 5. PARAMETER COMPARISON
# ---------------------------------------------------------
echo "5. PARAMETER COMPARISON"
echo "-----------------------"

# 5a. Sigma Comparison (Background Quality)
echo "a) Sigma Comparison (Background Quality)..."
# Low Sigma (High Compression)
echo "   - Low Sigma (0.1)"
python roi_compressor.py \
    --input data/images/car/0016cf15fa4d4e16.jpg \
    --output results/sigma_low.jpg \
    --checkpoint checkpoints/tic_lambda_0.0483.pth.tar \
    --sigma 0.1 \
    --seg-method yolo \
    --seg-classes car \
    --highlight

# High Sigma (Low Compression)
echo "   - High Sigma (0.9)"
python roi_compressor.py \
    --input data/images/car/0016cf15fa4d4e16.jpg \
    --output results/sigma_high.jpg \
    --checkpoint checkpoints/tic_lambda_0.0483.pth.tar \
    --sigma 0.9 \
    --seg-method yolo \
    --seg-classes car \
    --highlight

# 5b. Lambda Comparison (Rate-Distortion)
echo "b) Lambda Comparison (Rate-Distortion)..."
# Low Lambda
echo "   - Low Lambda (0.013)"
python roi_compressor.py \
    --input data/images/car/0016cf15fa4d4e16.jpg \
    --output results/lambda_low.jpg \
    --checkpoint checkpoints/tic_lambda_0.013.pth.tar \
    --sigma 0.3 \
    --lambda 0.013 \
    --N 128 \
    --M 192 \
    --seg-method yolo \
    --seg-classes car \
    --highlight

# High Lambda
echo "   - High Lambda (0.0932)"
python roi_compressor.py \
    --input data/images/car/0016cf15fa4d4e16.jpg \
    --output results/lambda_high.jpg \
    --checkpoint checkpoints/tic_lambda_0.0932.pth.tar \
    --sigma 0.3 \
    --lambda 0.0932 \
    --seg-method yolo \
    --seg-classes car \
    --highlight

echo
echo "---------------------------------------------------------"
echo

# -------------------------------------------------
# 5. Standalone Segmentation (Mask R-CNN)
# -------------------------------------------------
echo "--- Running Standalone Segmentation (Mask R-CNN) ---"
python roi_segmenter.py \
    --input images/car/0016cf15fa4d4e16.jpg \
    --output results/mask_rcnn.png \
    --method maskrcnn \
    --classes car \
    --visualize

# -------------------------------------------------
# 6. Video Compression (Static Mode)
# -------------------------------------------------
echo "--- Running Video Compression (Static Mode) ---"
python roi_compressor.py \
    --input data/videos/traffic.mp4 \
    --output results/video_static.mp4 \
    --quality-level 4 \
    --sigma 0.4 \
    --seg-method yolo \
    --seg-classes car person \
    --video-mode static \
    --output-fps 10

# -------------------------------------------------
# 7. Video Compression (Dynamic Mode)
# -------------------------------------------------
echo "--- Running Video Compression (Dynamic Mode) ---"
python roi_compressor.py \
    --input data/videos/traffic.mp4 \
    --output results/video_dynamic.mp4 \
    --quality-level 4 \
    --seg-method yolo \
    --seg-classes car person \
    --video-mode dynamic \
    --target-bandwidth-kbps 800 \
    --min-fps 5 \
    --max-fps 20

# -------------------------------------------------
# 8. Video Detection Evaluation (Static Mode)
# -------------------------------------------------
echo "--- Running Video Detection Evaluation (Static Mode) ---"
python roi_detection_eval.py \
    --before data/videos/traffic.mp4 \
    --video-mode static \
    --sigma 0.3 \
    --output-fps 10 \
    --quality-level 4 \
    --seg-method yolo \
    --seg-classes car person \
    --detectors yolo \
    --max-video-frames 30 \
    --video-sample-interval 3 \
    --save-after results/video_eval_compressed.mp4 \
    --out results/video_detection_eval.json \
    --viz-dir results/video_detection_viz

# -------------------------------------------------
# 9. Video Detection Evaluation (Comparing Two Videos)
# -------------------------------------------------
echo "--- Running Video Detection Evaluation (Compare Two Videos) ---"
python roi_detection_eval.py \
    --before data/videos/traffic.mp4 \
    --after results/video_static.mp4 \
    --detectors yolo \
    --max-video-frames 30 \
    --video-sample-interval 3 \
    --out results/video_compare_eval.json

echo
echo "============================="
echo "All examples complete! Check results/ directory"