|
Github开源项目
YOLOV8辅助对准
怎样设置环境:
pip install -r requirements.txt
假如您有支持 CUDA 的 GPU,则可以运行以下额外下令
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
利用 TensorRT 加快(可选)
添加以下下令:
pip install --upgrade setuptools pip --userpip install nvidia-pyindexpip install --upgrade nvidia-tensorrtpip install pycuda
假如你不能以这种方式安装tensorrt,你可以查找这个[color=var(--color-accent-fg)]Nvidia指南。
我提供了“.trt”模子,但很有大概您必须本身将“.pt”模子转换为“.trt”模子,由于 Tensorrt 引擎是特定于环境的。此存储库大概会有所资助: [color=var(--color-accent-fg)]TensorRT-For-YOLO-Series
怎样运行步伐:
只需利用以下下令运行文件main.py
python main.py
几秒钟后,步伐将开始运行。您可以在控制台中看到。Main Start
一旦你按住鼠标右键或鼠标左键(无论你按住对准照旧开始射击),步伐将开始对准对头。
怎样更改设置:
您可以更改文件中的设置。args.py
一些紧张的设置!!!:
模子
默认模子实用于 Apex。但是,您可以利用 练习本身的模子,并利用此设置切换模子。train.py
“模子”目次中有几个模子,您可以选择此中之一。
这些模子用于 tensorRT,它比模子快约 4 倍,但精度雷同。.trt.pt
模子速率:8n>8s>8m
模子精度:8n<8s<8m
crop_size:
此设置确定要检测的屏幕部门。太高大概会导致难以检测到小物体。
本帖潜伏的内容
https://pan.quark.cn/s/3ad32ecd2e56
https://pan.baidu.com/s/18md9mEhHA9M1d-beugfokA?pwd=a1dh 提取码:a1dh
|
|