·win + pycharm ·sahi==0.8.4 ·yolov5==5.0 ·pytorch==1.7.1+cu101
# 安装gpu版torch和torchvision
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
#
pip install yolov5
pip install sahi==0.8.4
pip install fiftyone imantics
-
tree:
main.py
README.md
samll-vehicles1.jpg
yolov5s6.pt -
vim main.py:
line4: yolov5_model_path = yolov5_model_path = "C:\\Users\\kiven\\Desktop\\SAHI\\yolov5s6.pt" # 放置你的权重地址
line10: device="cuda:0", # or 'cpu'
- please change your path and divice
- python main.py