Colaboratory 简称"Colab",是Google Research 团队开发的一款产品。 在Colab 中,任何人都可以通过浏览器编写和执行任意Python 代码。 它尤其适合机器学习、数据分析和教育目的。 从技术上说,Colab 是一种托管式Jupyter 笔记本服务。
大佬们写的工程源码在这里:https://github.com/hasantha-nirmal/Traffic_Violation_Detection_Yolov4_Deep-Sort
下载官方yolov4预训练权重模型,将 yolov4.weights 从您的下载文件夹复制并粘贴到此存储库的“数据”文件夹中。https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v3_optimal/yolov4.weights
** 1.**先将工程传到自己的谷歌云端硬盘上。
2.新建更多打开 Colaboratory
3.将谷歌云硬盘挂载到colab上并且修改当前路径到自己工程上。(path为自己需要修改的路径)
import os
from google.colab import drive
drive.mount('/content/drive')
path = "/content/drive/MyDrive/Traffic_Violation_Detection_Yolov4_Deep-Sort-main"
os.chdir(path)
print(os.getcwd())
4.通过 Anaconda 或 Pip 安装正确的依赖项。
5.conda 需要在colab上下载miniconda 因此我选择直接pip下载。(GPU相较于CPU运行速度更快)
!pip install -r requirement-gpu.txt
6.requirement-gpu.txt 根据 yml文件更改下:
opencv-python==4.1.1.26
lxml
tqdm
tensorflow==2.3.0
absl-py
easydict
matplotlib
pillow
7.执行以下命令将暗网权重转换为张流量模型。
!python save_model.py --model yolov4
8.运行以下命令执行停车违章检测程序。
!python parking_violation_detection.py --output ./outputs/processed_vids/parking.avi --model yolov4
--output 后面 ./outputs..... 将其设置为保存到“输出”文件夹
parking.avi 将video输出后的格式为 avi (AVI英文全称为Audio Video Interleaved,即音频视频交错格式)
9.后续执行过程仍然会遇到错误,例如:
X server是Linux系统上提供图形用户界面的服务程序。当客户端主机Client访问服务器Server上的图形程序时,需要Server对该Client赋能访问图形程序的权限。
后续解决方案:
打开 parking_violation_detection.py 将 cv2.imshow注释掉。
若有错误欢迎大家指正,后续操作上的问题可以评论区留言。
版权归原作者 小小小小小小菜! 所有, 如有侵权,请联系我们删除。