GithubHelp home page GithubHelp logo

wanghaoyuuu / yolov5-mask Goto Github PK

View Code? Open in Web Editor NEW
8.0 1.0 0.0 340.75 MB

YOLOv5是一种基于PyTorch的目标检测模型,可以快速准确地识别图像中的不同类别的物体。我们可以利用yolov5来实现一个是否佩戴口罩的预测系统。

Python 46.74% QML 52.99% JavaScript 0.22% Shell 0.03% Dockerfile 0.01%

yolov5-mask's Introduction

相关依赖包的安装

pip install -r requirements.txt

数据集为收集的不同的数据集统一整理后得到的

源码包含了yolov5的部分使用到的源码

GUI

visual.py为GUI文件,直接运行即可

是否启用GPU

使用之前可以改动是否启用GPU进行预测,

将self.device = 'cpu'中'cpu'改成'0' 即可使用GPU进行预测。

权重位置

训练好的权重文件位于 runs/train/exp3/weights/best.pt

如果不想使用GUI可以直接运行detect.py文件。

yolov5-mask's People

Contributors

wanghaoyuuu avatar

Stargazers

 avatar  avatar  avatar  avatar MoyerLiu avatar  avatar  avatar  avatar

Watchers

 avatar

yolov5-mask's Issues

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.