REM ģת
set YOLOV5_DIR=E:\src\yolov5
set "script_dir=%~dp0"

call D:\ProgramData\anaconda3\condabin\conda.bat activate yolov5
::pip install onnx
cd /D %YOLOV5_DIR%

python export.py --weights %script_dir%runs/train/exp2/weights/best.pt --img 640 --device cpu --include torchscript
move %script_dir%runs\train\exp2\weights\best.torchscript.pt %script_dir%libtorch.torchscript.cpu.pt

python export.py --weights %script_dir%runs/train/exp2/weights/best.pt --img 640 --device 0 --include torchscript
move %script_dir%runs\train\exp2\weights\best.torchscript.pt %script_dir%libtorch.torchscript.gpu.pt

echo "libtorchת"

pause