2021-10-09 16:20:50 索炜达电子 901
项目编号:E1454
文件大小:55M
源码说明:带中文注释
开发环境:Python
简要概述:
我们主要利用了三种计算机视觉算法,它们分别是:YOLOv5,SORT和Cascade-LD。
通过这些算法,我们实现了简易的违章检测,它们分别是:车辆压线检测、车辆超速检测和车辆逆行检测。
目录│文件列表:
└ UAV-traffic-patrol-main
│ main.py
│ reverse.py
├ Cascade_LD
│ │ CascadeLD.py
│ │ test1.jpg
│ │ test2.jpg
│ │ test2.png
│ │ __init__.py
│ ├ models
│ │ │ erfnet.py
│ │ └ lcnet.py
│ └ pretrained
│ │ classification_16_2class.pth
│ │ classification_16_3class.pth
│ │ classification_64_2class.pth
│ │ classification_64_3class.pth
│ └ erfnet_tusimple.pth
├ data
│ └ demo.mp4
├ output
│ │ demo.txt
│ └ 超速检测结果.mp4
└ yolov5_sort
│ detect.py
│ hubconf.py
│ sort.py
│ tempCodeRunnerFile.py
│ test.py
│ train.py
├ data
│ │ coco.yaml
│ │ coco128.yaml
│ │ hyp.finetune.yaml
│ │ hyp.scratch.yaml
│ │ voc.yaml
│ └ scripts
│ │ get_coco.sh
│ └ get_voc.sh
├ models
│ │ common.py
│ │ experimental.py
│ │ export.py
│ │ yolo.py
│ │ yolov5l.yaml
│ │ yolov5m.yaml
│ │ yolov5s.yaml
│ │ yolov5x.yaml
│ │ __init__.py
│ └ hub
│ │ anchors.yaml
│ │ yolov3-spp.yaml
│ │ yolov3-tiny.yaml
│ │ yolov3.yaml
│ │ yolov5-fpn.yaml
│ │ yolov5-p2.yaml
│ │ yolov5-p6.yaml
│ │ yolov5-p7.yaml
│ │ yolov5-panet.yaml
│ │ yolov5l6.yaml
│ │ yolov5m6.yaml
│ │ yolov5s6.yaml
│ └ yolov5x6.yaml
├ utils
│ │ activations.py
│ │ autoanchor.py
│ │ datasets.py
│ │ general.py
│ │ google_utils.py
│ │ loss.py
│ │ metrics.py
│ │ plots.py
│ │ torch_utils.py
│ │ __init__.py
│ ├ google_app_engine
│ │ │ additional_requirements.txt
│ │ │ app.yaml
│ │ └ Dockerfile
│ └ wandb_logging
│ │ log_dataset.py
│ │ wandb_utils.py
│ └ __init__.py
└ weights
└ download_weights.sh