-
Notifications
You must be signed in to change notification settings - Fork 482
Description
准备
下载安装 Anaconda
打开 Anaconda Prompt 控制台,在(base)主环境下
更新pip库 python.exe -m pip install --upgrade pip然后
安装 pip install opencv-contrib-python
- 配置一个独立的 虚拟环境
首先打开 Anaconda Prompt 控制台 或 Anaconda Navigator 可视化软件
创建一个虚拟环境 命名,如:SCM(这只是虚拟环境的名字)conda create -n SCM python=3.9.20Python版本你自己看着选,要符合 项目的版本 或 依赖库的版本 (很关键)
这个项目建议使用 Python3.9.20 (3.9.x)或以上版本(不支持最新版,至少3.12是不行的)
SafeConfigParser 类在 Python 3.2 及更高版本中已被弃用,并在 Python 3.8 中被完全移除
- 然后切换到虚拟环境
conda activate SCM
要运行程序先转到你的程序项目所在盘符 如 输入E: 或 D: 等,回车
然后cd到程序的目录 如: cd E:\YOLO\YOLO_v5\Test_1\Smart_Construction-master
更新 pip python.exe -m pip install --upgrade pip
更新 setuptools pip install --upgrade setuptools
-
替换 requirements.txt 文件,以便下载正确版本的依赖库
安装问题 #90 来源:安装问题 安装问题 #90 [Tanhongyun](https://github.com/Tanhongyun)大佬absl-py==0.12.0
altgraph==0.17
cachetools==4.2.1
certifi==2020.12.5
chardet==4.0.0
cycler==0.10.0
future==0.18.2
google-auth==1.28.1
google-auth-oauthlib==0.4.4
GPUtil==1.4.0
grpcio==1.37.0
idna==2.10
importlib-metadata==4.6
kiwisolver==1.3.1
Markdown==3.3.4
matplotlib==3.3.4
mkl-fft==1.3.6
mkl-random==1.2.2
mkl-service==2.4.0
numpy
oauthlib==3.1.0
olefile==0.46
opencv-python==4.5.1.48
Pillow==8.1.2
pefile==2022.5.30
protobuf==3.15.8
pyasn1==0.4.8
pyasn1-modules==0.2.8
pyinstaller-hooks-contrib==2024.3
pyparsing==2.4.7
PyQt5==5.15.4
PyQt5-Qt5==5.15.2
PyQt5-sip==12.8.1
PyQtChart==5.15.4
PyQtChart-Qt5==5.15.2
python-dateutil==2.8.1
pywin32-ctypes==0.2.1
PyYAML==5.4.1
requests==2.25.1
requests-oauthlib==1.3.0
rsa==4.7.2
scipy==1.6.1
six==1.15.0
tensorboard==2.4.1
tensorboard-plugin-wit==1.8.0
torch==1.8.0
torchvision==0.9.0
tornado==6.1
tqdm==4.59.0
typing-extensions==3.7.4.3
urllib3==1.26.5
Werkzeug==1.0.1
wincertstore==0.2
zipp==3.4.1
pyinstaller ==6.5.0
在 conda 虚拟环境安装依赖库 pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/
- 安装数据集
[GitHub - njvisionpower/Safety-Helmet-Wearing-Dataset: Safety helmet wearing detect dataset, with pretrained model](https://github.com/njvisionpower/Safety-Helmet-Wearing-Dataset)
把下载的数据集放在 ./data/Safety-Helmet-Wearing-Dataset-master/image或其他位置(建议在项目文件夹内)
- 安装权重文件
[百度网盘 请输入提取码](https://pan.baidu.com/s/1iMZkRNXY1fowpQCcapFDqw)
yolov5l 为基础训练 提取码: a66e
- 该项目作者提供的其他权重文件下载 详见该项目的Github网站
权重文件一般放在 weights文件夹内
Anaconda Prompt 控制台 运行
python detect.py --source ./Dataset/Safety-Helmet-Wearing-Dataset-master/image/ --weights weights/helmet_head_person_l.pt --conf 0.4(数据集目录、权重文件目录、置信度 按自己情况调整)
结果一般会保存在 .\inference\output中