Code Monkey home page Code Monkey logo

tyolov5's Introduction

TYolov5: A Temporal Yolov5 Detector Based on Quasi-Recurrent Neural Networks for Real-Time Handgun Detection in Video

Timely handgun detection is a crucial problem to improve public safety; nevertheless, the effectiveness of many surveillance systems still depends of finite human attention. Much of the previous research on handgun detection is based on static image detectors, leaving aside valuable temporal information that could be used to improve object detection in videos. To improve the performance of surveillance systems, a real-time temporal handgun detection system should be built. Using Temporal Yolov5, an architecture based on Quasi-Recurrent Neural Networks, temporal information is extracted from video to improve the results of handgun detection. Moreover, two publicly available datasets are proposed, labeled with hands, guns, and phones. One containing 2199 static images to train static detectors, and another with 5960 frames of videos to train temporal modules. Additionally, we explore two temporal data augmentation techniques based on Mosaic and Mixup. The resulting systems are three temporal architectures: one focused in reducing inference with a mAP50:95 of 55.9, another in having a good balance between inference and accuracy with a mAP50:95 of 59, and a last one specialized in accuracy with a mAP50:95 of 60.2. Temporal Yolov5 achieves real-time detection in the small and medium architectures. Moreover, it takes advantage of temporal features contained in videos to perform better than Yolov5 in our temporal dataset, making TYolov5 suitable for real-world applications.

Weights of the models:

S_ConvLSTM https://drive.google.com/file/d/1AO920AaKX18Ag22E7pmEvoWFZsr2wa6R/view?usp=sharing

M_ConvLSTM https://drive.google.com/file/d/120liXvIP71emZeC04_JtLW8ul4DBL8jI/view?usp=sharing

L_ConvLSTM https://drive.google.com/file/d/1-sn5QH4SsrooGvyiz9kHwYxsJdpJWpvF/view?usp=sharing

S_QRNN https://drive.google.com/file/d/11o10p30ViHBAlv2WqPX-RLKXZRTSf2XH/view?usp=sharing

M_QRNN https://drive.google.com/file/d/117ImSTS7bh4phX_85IAe0sVu0kzVYinz/view?usp=sharing

L_QRNN https://drive.google.com/file/d/108MaX8mFBVke_7-6qgrXb_9LrOo5U73Y/view?usp=sharing

If you use this code for your research, please consider citing:

Mario Alberto Duran-Vega, Miguel Gonzalez-Mendoza, Leonardo Chang, Cuauhtemoc Daniel Suarez-Ramirez https://arxiv.org/abs/2111.08867

tyolov5's People

Contributors

marioduran avatar

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.