2021-10-09 16:20:50 索煒達電子 898
項目編號:E1454
文件大?。?5M
源碼說明:帶中文注釋
開發(fā)環(huán)境:Python
簡要概述:
我們主要利用了三種計算機視覺算法,它們分別是:YOLOv5,SORT和Cascade-LD。
通過這些算法,我們實現(xiàn)了簡易的違章檢測,它們分別是:車輛壓線檢測、車輛超速檢測和車輛逆行檢測。
目錄│文件列表:
└ 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