YaphetS-X / CenterNet-MobileNetV3

CenterNet with MobileNetV3 backboned helmet detection based on PyTorch with inference code only.

MIT License

51 stars 18 forks source link

readme

A PyTorch implementation of CenterNet and MobileNetV3

Introduction

CenterNet with MobileNetV3 backboned helmet detection based on PyTorch with inference code only.

This repo is forked from CenterNet and MobileNetV3.

result

Installation

cd $ROOT/lib/models/networks/DCNv2
python setup.py build develop

Usage

python demo.py