File size: 4,518 Bytes
36fcf33 | 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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 | @echo off
REM Batch script to check if all dependencies are installed
REM Run this script from the project root directory
echo =========================================
echo Dependency Checker for SAM2 Image Auto Segment
echo =========================================
echo.
REM Check if Python is installed
python --version >nul 2>&1
if errorlevel 1 (
echo [ERROR] Python is not installed or not in PATH!
echo Please install Python 3.10 or higher
pause
exit /b 1
)
python --version
echo.
echo Checking installed packages...
echo.
REM List of required packages from requirements.txt
set MISSING_PACKAGES=
set FAILED_IMPORTS=
REM Check each package
echo [1/14] Checking fastapi...
python -c "import fastapi; print(' OK - fastapi', fastapi.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% fastapi
echo [2/14] Checking uvicorn...
python -c "import uvicorn; print(' OK - uvicorn', uvicorn.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% uvicorn
echo [3/14] Checking numpy...
python -c "import numpy; print(' OK - numpy', numpy.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% numpy
echo [4/14] Checking opencv-python...
python -c "import cv2; print(' OK - opencv-python', cv2.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% opencv-python
echo [5/14] Checking Pillow...
python -c "import PIL; print(' OK - Pillow', PIL.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% Pillow
echo [6/14] Checking scikit-image...
python -c "import skimage; print(' OK - scikit-image', skimage.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% scikit-image
echo [7/14] Checking torch...
python -c "import torch; print(' OK - torch', torch.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% torch
echo [8/14] Checking torchvision...
python -c "import torchvision; print(' OK - torchvision', torchvision.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% torchvision
echo [9/14] Checking huggingface_hub...
python -c "import huggingface_hub; print(' OK - huggingface_hub', huggingface_hub.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% huggingface_hub
echo [10/14] Checking omegaconf...
python -c "import omegaconf; print(' OK - omegaconf', omegaconf.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% omegaconf
echo [11/14] Checking hydra-core...
python -c "import hydra; print(' OK - hydra-core', hydra.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% hydra-core
echo [12/14] Checking iopath...
python -c "import iopath; print(' OK - iopath')" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% iopath
echo [13/14] Checking tqdm...
python -c "import tqdm; print(' OK - tqdm', tqdm.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% tqdm
echo [14/14] Checking requests...
python -c "import requests; print(' OK - requests', requests.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% requests
echo [15/14] Checking psutil...
python -c "import psutil; print(' OK - psutil', psutil.__version__)" 2>nul || set MISSING_PACKAGES=%MISSING_PACKAGES% psutil
echo.
echo Checking SAM2 package...
python -c "import sam2; print(' OK - sam2 package is installed')" 2>nul || (
echo [WARNING] SAM2 package is not installed
echo Run: cd sam2 ^&^& python -m pip install -e . ^&^& cd ..
set MISSING_PACKAGES=%MISSING_PACKAGES% sam2
)
echo.
echo =========================================
if "%MISSING_PACKAGES%"=="" (
echo All dependencies are installed! ✓
echo.
echo Testing critical imports...
python -c "from model.sam_model import predict_polygon; print(' OK - model.sam_model')" 2>nul || echo [WARNING] Cannot import model.sam_model
python -c "from model.utils import load_image_from_url; print(' OK - model.utils')" 2>nul || echo [WARNING] Cannot import model.utils
python -c "from sam2.automatic_mask_generator import SAM2AutomaticMaskGenerator; print(' OK - sam2.automatic_mask_generator')" 2>nul || echo [WARNING] Cannot import sam2.automatic_mask_generator
echo.
echo ✓ Ready to run the application!
) else (
echo Missing packages:%MISSING_PACKAGES%
echo.
echo Please install missing packages by running:
echo install_requirements.bat
echo.
echo Or manually:
echo python -m pip install -r requirements.txt
echo cd sam2
echo python -m pip install -e .
echo cd ..
)
echo =========================================
echo.
pause
|