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
49 lines
1.5 KiB
Batchfile
49 lines
1.5 KiB
Batchfile
@echo off
|
|
REM ComfyUI YOLOv8 Detection Node Installation Script for Windows
|
|
|
|
echo Installing ComfyUI YOLOv8 Detection Node...
|
|
|
|
REM Check if pip is available
|
|
pip --version >nul 2>&1
|
|
if %errorlevel% neq 0 (
|
|
echo Error: pip is not installed. Please install Python and pip first.
|
|
pause
|
|
exit /b 1
|
|
)
|
|
|
|
REM Install required packages
|
|
echo Installing required Python packages...
|
|
pip install -r requirements.txt
|
|
|
|
if %errorlevel% neq 0 (
|
|
echo Error: Failed to install some packages. Please check your Python environment.
|
|
pause
|
|
exit /b 1
|
|
)
|
|
|
|
REM Check if ultralytics is properly installed
|
|
python -c "import ultralytics; print('✓ ultralytics installed successfully')" 2>nul
|
|
if %errorlevel% neq 0 (
|
|
echo Error: Failed to install ultralytics. Trying alternative installation...
|
|
pip install ultralytics --upgrade
|
|
)
|
|
|
|
REM Download default YOLOv8 model (optional)
|
|
echo Downloading default YOLOv8 nano model...
|
|
python -c "from ultralytics import YOLO; model = YOLO('yolov8n.pt'); print('✓ YOLOv8 nano model downloaded successfully')" 2>nul
|
|
if %errorlevel% neq 0 (
|
|
echo Warning: Could not download model. Model will be downloaded automatically on first use.
|
|
)
|
|
|
|
echo.
|
|
echo Installation completed!
|
|
echo.
|
|
echo To use this node in ComfyUI:
|
|
echo 1. Copy this folder to your ComfyUI/custom_nodes/ directory
|
|
echo 2. Restart ComfyUI
|
|
echo 3. Look for 'YOLOv8 Object Detection' in the image/object_detection category
|
|
echo.
|
|
echo For more information, see the README.md file.
|
|
echo.
|
|
pause
|