GitPlanet

Cheap and reliable Node.js hosting starts at $3/month, and $1/month static HTML hosting

Created with love in Canada, visit hostnodejs.com today

Feel like to post an Ad? Learn Details
All Projects → shaoshengsong → Mobilenetv3 Ssd

shaoshengsong / Mobilenetv3 Ssd

MobileNetV3-SSD for object detection and implementation in PyTorch

Programming Languages

python
139335 projects - #7 most used programming language

Labels

onnxmobilenetssd

Projects that are alternatives of or similar to Mobilenetv3 Ssd

Fast Stacked Hourglass Network OpenVino
A fast stacked hourglass network for human pose estimation on OpenVino
Stars: ✭ 52 (-72.34%)
Mutual labels:  mobilenet, onnx
Proctoring Ai
Creating a software for automatic monitoring in online proctoring
Stars: ✭ 155 (-17.55%)
Mutual labels:  ssd, mobilenet
MobileNet-SSD-windows
No description or website provided.
Stars: ✭ 91 (-51.6%)
Mutual labels:  ssd, mobilenet
MobileNetV3-SSD-Compact-Version
MobileNetV3 SSD的简洁版本
Stars: ✭ 65 (-65.43%)
Mutual labels:  ssd, mobilenet
Tensorflow Face Detection
A mobilenet SSD based face detector, powered by tensorflow object detection api, trained by WIDERFACE dataset.
Stars: ✭ 711 (+278.19%)
Mutual labels:  ssd, mobilenet
Php Opencv Examples
Tutorial for computer vision and machine learning in PHP 7/8 by opencv (installation + examples + documentation)
Stars: ✭ 333 (+77.13%)
Mutual labels:  onnx, mobilenet
hand detection
A Light CNN based Method for Hand Detection and Orientation Estimation
Stars: ✭ 116 (-38.3%)
Mutual labels:  ssd, mobilenet
Tf Object Detection
Simpler app for tensorflow object detection API
Stars: ✭ 91 (-51.6%)
Mutual labels:  ssd, mobilenet
Ssds.pytorch
Repository for Single Shot MultiBox Detector and its variants, implemented with pytorch, python3.
Stars: ✭ 570 (+203.19%)
Mutual labels:  ssd, mobilenet
Mobilenetv2 Ssdlite
Caffe implementation of SSD and SSDLite detection on MobileNetv2, converted from tensorflow.
Stars: ✭ 435 (+131.38%)
Mutual labels:  ssd, mobilenet
Ssds pytorch
Multiple basenet MobileNet v1,v2, ResNet combined with SSD detection method and it's variants such as RFB, FSSD etc.
Stars: ✭ 71 (-62.23%)
Mutual labels:  ssd, mobilenet
Mobilenet Ssd
Caffe implementation of Google MobileNet SSD detection network, with pretrained weights on VOC0712 and mAP=0.727.
Stars: ✭ 1,805 (+860.11%)
Mutual labels:  ssd, mobilenet
Nngen
NNgen: A Fully-Customizable Hardware Synthesis Compiler for Deep Neural Network
Stars: ✭ 149 (-20.74%)
Mutual labels:  onnx
Handpose
A python program to detect and classify hand pose using deep learning techniques
Stars: ✭ 168 (-10.64%)
Mutual labels:  ssd
Caffe Mobilenet
A caffe implementation of mobilenet's depthwise convolution layer.
Stars: ✭ 146 (-22.34%)
Mutual labels:  mobilenet
Ncnn
ncnn is a high-performance neural network inference framework optimized for the mobile platform
Stars: ✭ 13,376 (+7014.89%)
Mutual labels:  onnx
Vip
Video Platform for Action Recognition and Object Detection in Pytorch
Stars: ✭ 175 (-6.91%)
Mutual labels:  ssd
A Pytorch Tutorial To Object Detection
SSD: Single Shot MultiBox Detector | a PyTorch Tutorial to Object Detection
Stars: ✭ 2,398 (+1175.53%)
Mutual labels:  ssd
Efficientnet
Implementation of EfficientNet model. Keras and TensorFlow Keras.
Stars: ✭ 1,920 (+921.28%)
Mutual labels:  mobilenet
Posenet Coreml
I checked the performance by running PoseNet on CoreML
Stars: ✭ 143 (-23.94%)
Mutual labels:  mobilenet
View All Similar Projects ➔

MobileNetV3-SSD

MobileNetV3-SSD implementation in PyTorch

关于第二个版本请移步 https://github.com/shaoshengsong/MobileNetV3-SSD-Compact-Version 有测试结果 希望尝试新技术请到这里 https://github.com/shaoshengsong/quarkdet 一个轻量级的目标检测包括多种模型 目的 Object Detection 应用于目标检测

环境

操作系统: Ubuntu18.04

Python: 3.6

PyTorch: 1.1.0

使用MobileNetV3-SSD实现目标检测

Support Export ONNX

代码参考（严重参考以下代码）

一 SSD部分

A PyTorch Implementation of Single Shot MultiBox Detector

二 MobileNetV3 部分

1 mobilenetv3 with pytorch，provide pre-train model

2 MobileNetV3 in pytorch and ImageNet pretrained models

3Implementing Searching for MobileNetV3 paper using Pytorch

4 MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv. no discernible latency cost.

针对4我这里没有做MobileNetV1, MobileNetV2等等代码兼容，只有MobileNetV3可用

下载数据 本例是以蛋糕和面包为例，原因是数据量小 所有类别总大小是561G，蛋糕和面包是3.2G

python3 open_images_downloader.py --root /media/santiago/a/data/open_images --class_names "Cake,Bread" --num_workers 20

训练过程

首次训练

python3 train_ssd.py --dataset_type open_images --datasets /media/santiago/data/open_images --net mb3-ssd-lite --scheduler cosine --lr 0.01 --t_max 100 --validation_epochs 5 --num_epochs 100 --base_net_lr 0.001 --batch_size 5

预加载之前训练的模型

python3 train_ssd.py --dataset_type open_images --datasets /media/santiago/data/open_images --net mb3-ssd-lite --pretrained_ssd models/mb3-ssd-lite-Epoch-99-Loss-2.5194434596402613.pth --scheduler cosine --lr 0.01 --t_max 100 --validation_epochs 5 --num_epochs 200 --base_net_lr 0.001 --batch_size 5

测试一张图片

python run_ssd_example.py mb3-ssd-lite models/mb3-ssd-lite-Epoch-99-Loss-2.5194434596402613.pth models/open-images-model-labels.txt /home/santiago/picture/test.jpg

视频检测

python3 run_ssd_live_demo.py mb3-ssd-lite models/mb3-ssd-lite-Epoch-99-Loss-2.5194434596402613.pth models/open-images-model-labels.txt

Cake and Bread Pretrained model

链接: https://pan.baidu.com/s/1byY1eJk3Hm3CTp-29KirxA

提取码: qxwv

VOC Dataset Pretrained model

链接: https://pan.baidu.com/s/1yt_IRY0RcgSxB-YwywoHuA

提取码: 2sta

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].