CodeProject.AI-ObjectDetectionCoral/install.bat at main · codeproject/CodeProject.AI-ObjectDetectionCoral · GitHub

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
:: Installation script :::::::::::::::::::::::::::::::::::::::::::::::::::::::::
::
::                         ObjectDetection (Coral)
::
:: This script is only called from ..\..\CodeProject.AI-Server\src\setup.bat in
:: Dev setup, or ..\..\src\setup.bat in production
::
:: For help with install scripts, notes on variables and methods available, tips,
:: and explanations, see CodeProject.AI-Server\src\modules\install_script_help.md

@if "%1" NEQ "install" (
    echo This script is only called from ..\..\src\setup.bat
    @pause
    @goto:eof
)

REM Backwards compatibility with Server 2.6.5
if "!utilsScript!" == "" if "!sdkScriptsDirPath!" NEQ "" set utilsScript=%sdkScriptsDirPath%\utils.bat

REM Do this in requirements file. Otherwise we need the TFLite wheel too
REM if "%pythonVersion%" == "3.7" (
REM     set "wheel=https://github.com/google-coral/pycoral/releases/download/v2.0.0/pycoral-2.0.0-cp37-cp37m-win_amd64.whl"
REM ) else if "%pythonVersion%" == "3.8" (
REM     set "wheel=https://github.com/google-coral/pycoral/releases/download/v2.0.0/pycoral-2.0.0-cp38-cp38-win_amd64.whl"
REM ) else if "%pythonVersion%" == "3.9" (
REM     set "wheel=https://github.com/google-coral/pycoral/releases/download/v2.0.0/pycoral-2.0.0-cp39-cp39-win_amd64.whl"
REM )
REM call "!utilsScript!" InstallPythonPackagesByName %wheel% "PyCoral API"

:: Install supporting Libraries
if not exist edgetpu_runtime (
    REM edgetpu_runtime_20221024 is badly zipped. edgetpu_runtime-20221024 is better
    REM call "%utilsScript%" GetFromServer "libraries/" "edgetpu_runtime_20221024.zip" "." "Downloading edge TPU runtime..."
    call "%utilsScript%" GetFromServer "libraries/" "edgetpu_runtime-20221024.zip" "edgetpu_runtime" "Downloading edge TPU runtime..."
)
if exist edgetpu_runtime (

    net session > NUL
    IF %ERRORLEVEL% EQU 0 (
        call "!utilsScript!" WriteLine "Installing the edge TPU libraries..." "!color_info!"
        pushd edgetpu_runtime
        call install.bat
        popd

        REM TODO: Check the edge TPU install worked
        REM set moduleInstallErrors=...
    ) else (
        call "!utilsScript!" WriteLine "=================================================================================" "!color_info!"
        call "!utilsScript!" WriteLine "Please run !moduleDirPath!\edgetpu_runtime\install.bat to complete this process." "!color_info!"
        call "!utilsScript!" WriteLine "=================================================================================" "!color_info!"
    )
)

:: Download the TFLite/edgeTPU models and store in /assets
:: call "%utilsScript%" GetFromServer "models/" "objectdetect-coral-multitpu-models.zip" "assets" "Downloading MobileNet models..."

call "%utilsScript%" GetFromServer "models/" "objectdetection-efficientdet-large-edgetpu.zip" "assets" "Downloading EfficientDet (large) models..."
call "%utilsScript%" GetFromServer "models/" "objectdetection-efficientdet-medium-edgetpu.zip" "assets" "Downloading EfficientDet (medium) models..."
call "%utilsScript%" GetFromServer "models/" "objectdetection-efficientdet-small-edgetpu.zip" "assets" "Downloading EfficientDet (small) models..."
call "%utilsScript%" GetFromServer "models/" "objectdetection-efficientdet-tiny-edgetpu.zip" "assets" "Downloading EfficientDet (tiny) models..."

call "%utilsScript%" GetFromServer "models/" "objectdetection-mobilenet-large-edgetpu.zip" "assets" "Downloading MobileNet (large) models..."
call "%utilsScript%" GetFromServer "models/" "objectdetection-mobilenet-medium-edgetpu.zip" "assets" "Downloading MobileNet (medium) models..."
call "%utilsScript%" GetFromServer "models/" "objectdetection-mobilenet-small-edgetpu.zip" "assets" "Downloading MobileNet (small) models..."
call "%utilsScript%" GetFromServer "models/" "objectdetection-mobilenet-tiny-edgetpu.zip" "assets" "Downloading MobileNet (tiny) models..."

:: call "%utilsScript%" GetFromServer "models/" "objectdetection-yolov5-large-edgetpu.zip" "assets" "Downloading YOLOv5 (large) models..."
:: call "%utilsScript%" GetFromServer "models/" "objectdetection-yolov5-medium-edgetpu.zip" "assets" "Downloading YOLOv5 (medium) models..."
:: call "%utilsScript%" GetFromServer "models/" "objectdetection-yolov5-small-edgetpu.zip" "assets" "Downloading YOLOv5 (small) models..."
:: call "%utilsScript%" GetFromServer "models/" "objectdetection-yolov5-tiny-edgetpu.zip" "assets" "Downloading YOLOv5 (tiny) models..."

call "%utilsScript%" GetFromServer "models/" "objectdetection-yolov8-large-edgetpu.zip" "assets" "Downloading YOLOv8 (large) models..."
call "%utilsScript%" GetFromServer "models/" "objectdetection-yolov8-medium-edgetpu.zip" "assets" "Downloading YOLOv8 (medium) models..."
call "%utilsScript%" GetFromServer "models/" "objectdetection-yolov8-small-edgetpu.zip" "assets" "Downloading YOLOv8 (small) models..."
call "%utilsScript%" GetFromServer "models/" "objectdetection-yolov8-tiny-edgetpu.zip" "assets" "Downloading YOLOv8 (tiny) models..."

REM TODO: Check assets created and has files
REM set moduleInstallErrors=...