pswdet_deploy

道路抛洒物检测(YOLOv8 -> PIDNet -> ResNet)

https://github.com/dirtybit64/pswdet_deploy

Science Score: 44.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • CITATION.cff file
    Found CITATION.cff file
  • codemeta.json file
    Found codemeta.json file
  • .zenodo.json file
    Found .zenodo.json file
  • DOI references
  • Academic publication links
  • Academic email domains
  • Institutional organization owner
  • JOSS paper metadata
  • Scientific vocabulary similarity
    Low similarity (1.7%) to scientific vocabulary
Last synced: 6 months ago · JSON representation ·

Repository

道路抛洒物检测(YOLOv8 -> PIDNet -> ResNet)

Basic Info
  • Host: GitHub
  • Owner: DirtyBit64
  • Language: Python
  • Default Branch: master
  • Homepage:
  • Size: 310 KB
Statistics
  • Stars: 11
  • Watchers: 2
  • Forks: 2
  • Open Issues: 1
  • Releases: 0
Created almost 2 years ago · Last pushed 7 months ago
Metadata Files
Readme Contributing Citation

README.md

pswdet_deploy

道路抛洒物检测 ———— 基于仓库YOLOv8进行订制化开发,针对抛洒物检测误报过多等问题,嵌入分割过滤模块PIDNet(用于排除道路区域外的误报)和分类过滤模块ResNet(用于排除道路内的误报)。

Install

bash pip install -r requirements.txt

:label: TODO

  • [x] 添加分割过滤逻辑
  • [x] 添加分类过滤逻辑
  • [x] 添加时序过滤逻辑

weight

通过网盘分享的文件:pswdet.zip 链接: https://pan.baidu.com/s/19SeaxRw4KzLMsIMG6kPzrQ?pwd=3456 提取码: 3456 复制这段内容后打开百度网盘手机App,操作更方便哦

Extended arguments

  • seg_filter : 是否启用分割过滤模块,若开启需将道路分割模型权重如pidnet.onnx置于ultralytics/models/yolo/detect/roadseg/*.onnx路径下
  • cls_filter : 是否启用分类过滤模块,若开启需将分类模型权重如psw64a.onnx置于ultralytics/models/yolo/detect/clsfilter/*.onnx路径下
  • save_if_box_father : 仅当save=false时,开启该参数有效,实现只保存检测出目标的结果集
  • frame_iou : 用于数据源为视频或流时,当前帧检测框与上一个帧检测框进行iou过滤,以减少时序重复检测,默认阈值为0

Contributing

PRs accepted.

License

nuist © ailab-cv

Owner

  • Name: DirtyBit
  • Login: DirtyBit64
  • Kind: user

Citation (CITATION.cff)

cff-version: 1.2.0
preferred-citation:
  type: software
  message: If you use this software, please cite it as below.
  authors:
  - family-names: Jocher
    given-names: Glenn
    orcid: "https://orcid.org/0000-0001-5950-6979"
  - family-names: Chaurasia
    given-names: Ayush
    orcid: "https://orcid.org/0000-0002-7603-6750"
  - family-names: Qiu
    given-names: Jing
    orcid: "https://orcid.org/0000-0003-3783-7069"
  title: "YOLO by Ultralytics"
  version: 8.0.0
  # doi: 10.5281/zenodo.3908559  # TODO
  date-released: 2023-1-10
  license: AGPL-3.0
  url: "https://github.com/ultralytics/ultralytics"

GitHub Events

Total
  • Issues event: 5
  • Watch event: 10
  • Issue comment event: 12
  • Push event: 2
  • Fork event: 1
Last Year
  • Issues event: 5
  • Watch event: 10
  • Issue comment event: 12
  • Push event: 2
  • Fork event: 1

Issues and Pull Requests

Last synced: 6 months ago

All Time
  • Total issues: 2
  • Total pull requests: 0
  • Average time to close issues: 3 days
  • Average time to close pull requests: N/A
  • Total issue authors: 2
  • Total pull request authors: 0
  • Average comments per issue: 5.0
  • Average comments per pull request: 0
  • Merged pull requests: 0
  • Bot issues: 0
  • Bot pull requests: 0
Past Year
  • Issues: 2
  • Pull requests: 0
  • Average time to close issues: 3 days
  • Average time to close pull requests: N/A
  • Issue authors: 2
  • Pull request authors: 0
  • Average comments per issue: 5.0
  • Average comments per pull request: 0
  • Merged pull requests: 0
  • Bot issues: 0
  • Bot pull requests: 0
Top Authors
Issue Authors
  • lonxiah (1)
  • tanglinjie0815 (1)
Pull Request Authors
Top Labels
Issue Labels
Pull Request Labels

Dependencies

docker/Dockerfile docker
  • pytorch/pytorch 2.0.1-cuda11.7-cudnn8-runtime build
requirements.txt pypi
  • matplotlib >=3.2.2
  • numpy >=1.22.2
  • onnxruntime *
  • opencv-python *
  • pandas >=1.1.4
  • pillow >=7.1.2
  • psutil *
  • py-cpuinfo *
  • pyyaml >=5.3.1
  • requests >=2.23.0
  • scipy >=1.4.1
  • seaborn >=0.11.0
  • setuptools *
  • thop *
  • torch >=1.8.0
  • torchvision >=0.9.0
  • tqdm >=4.64.0
  • typing *
  • wandb *
setup.py pypi