Eduardo Ottone
ee92581211
- Implements YOLOv8-based object detection with batch output - Supports multiple YOLOv8 model sizes (nano to extra-large) - Configurable confidence threshold, IoU threshold, and padding - Outputs separate images for each detected object - Includes installation scripts for Windows, Linux, and Mac - Complete documentation and example workflow - MIT license for open source distribution
64 lines
1.6 KiB
Bash
Executable File
64 lines
1.6 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Setup script for AI Object Segmentation Suite
|
|
echo "================================"
|
|
echo "AI Object Segmentation Suite Setup"
|
|
echo "================================"
|
|
|
|
# Check Python version
|
|
python_version=$(python3 --version 2>&1 | grep -Po '(?<=Python )\d+\.\d+')
|
|
echo "✓ Found Python $python_version"
|
|
|
|
# Create virtual environment
|
|
echo ""
|
|
echo "Creating virtual environment..."
|
|
if [ ! -d "venv" ]; then
|
|
python3 -m venv venv
|
|
echo "✓ Virtual environment created"
|
|
else
|
|
echo "✓ Virtual environment already exists"
|
|
fi
|
|
|
|
# Activate virtual environment
|
|
echo ""
|
|
echo "Activating virtual environment..."
|
|
source venv/bin/activate
|
|
|
|
# Upgrade pip
|
|
echo ""
|
|
echo "Upgrading pip..."
|
|
pip install --upgrade pip --quiet
|
|
|
|
# Install requirements
|
|
echo ""
|
|
echo "Installing requirements..."
|
|
echo "This may take a few minutes..."
|
|
pip install -r requirements.txt
|
|
|
|
# Download sample image for testing
|
|
echo ""
|
|
echo "Downloading sample test image..."
|
|
if [ ! -f "sample_test.jpg" ]; then
|
|
curl -s -o sample_test.jpg "https://images.unsplash.com/photo-1529156069898-49953e39b3ac?w=800"
|
|
echo "✓ Sample image downloaded as 'sample_test.jpg'"
|
|
else
|
|
echo "✓ Sample image already exists"
|
|
fi
|
|
|
|
# Create output directories
|
|
echo ""
|
|
echo "Creating output directories..."
|
|
mkdir -p output
|
|
|
|
echo ""
|
|
echo "================================"
|
|
echo "Setup Complete!"
|
|
echo "================================"
|
|
echo ""
|
|
echo "To get started:"
|
|
echo "1. Activate the environment: source venv/bin/activate"
|
|
echo "2. Test with sample image: python segment_practical.py sample_test.jpg"
|
|
echo "3. Check results in: practical_output/"
|
|
echo ""
|
|
echo "For more options, see README.md"
|