All Projects → talebolano → Yolov3 Network Slimming

talebolano / Yolov3 Network Slimming

yolov3 network slimming剪枝的一种实现

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Yolov3 Network Slimming

Yolov3v4 Modelcompression Multidatasettraining Multibackbone
YOLO ModelCompression MultidatasetTraining
Stars: ✭ 287 (-10.31%)
Mutual labels:  yolo, pruning
Ridurre Network Filter Pruning Keras
Keras model convolutional filter pruning package
Stars: ✭ 35 (-89.06%)
Mutual labels:  pruning, network
sparsezoo
Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes
Stars: ✭ 264 (-17.5%)
Mutual labels:  yolo, pruning
Jeelizar
JavaScript object detection lightweight library for augmented reality (WebXR demos included). It uses convolutional neural networks running on the GPU with WebGL.
Stars: ✭ 296 (-7.5%)
Mutual labels:  network
Yolo Pytorch
YOLO for object detection tasks
Stars: ✭ 301 (-5.94%)
Mutual labels:  yolo
Nxviz
Visualization Package for NetworkX
Stars: ✭ 307 (-4.06%)
Mutual labels:  network
Moon
A cross-platform,lightweight,scalable game server framework written in C++, and support Lua Script
Stars: ✭ 313 (-2.19%)
Mutual labels:  network
Ngraph.graph
Graph data structure in JavaScript
Stars: ✭ 295 (-7.81%)
Mutual labels:  network
React Native Network Info
React Native library for getting information about the devices network
Stars: ✭ 313 (-2.19%)
Mutual labels:  network
V Offline
🔌 Simple VueJS component to detect offline & online changes.
Stars: ✭ 307 (-4.06%)
Mutual labels:  network
Ccat
Cisco Config Analysis Tool
Stars: ✭ 305 (-4.69%)
Mutual labels:  network
Trojan Go
Go实现的Trojan代理,支持多路复用/路由功能/CDN中转/Shadowsocks混淆插件,多平台,无依赖。A Trojan proxy written in Go. An unidentifiable mechanism that helps you bypass GFW. https://p4gefau1t.github.io/trojan-go/
Stars: ✭ 4,049 (+1165.31%)
Mutual labels:  network
Social Distancing Analyser Covid 19
A social distancing analyzer AI tool to regulate social distancing protocol using video surveillance of CCTV cameras and drones. Social Distancing Analyser to prevent COVID19
Stars: ✭ 310 (-3.12%)
Mutual labels:  yolo
Pink
A high performance network library
Stars: ✭ 297 (-7.19%)
Mutual labels:  network
Frontendwingman
Frontend Wingman, Learn frontend faster!
Stars: ✭ 315 (-1.56%)
Mutual labels:  network
Deep Sort Yolov4
People detection and optional tracking with Tensorflow backend.
Stars: ✭ 306 (-4.37%)
Mutual labels:  yolo
Tensorflow 2.x Yolov3
YOLOv3 implementation in TensorFlow 2.3.1
Stars: ✭ 300 (-6.25%)
Mutual labels:  yolo
Yolo V2 Pytorch
YOLO for object detection tasks
Stars: ✭ 302 (-5.62%)
Mutual labels:  yolo
Alturos.yolo
C# Yolo Darknet Wrapper (real-time object detection)
Stars: ✭ 308 (-3.75%)
Mutual labels:  yolo
S task
awaitable coroutine library for C
Stars: ✭ 317 (-0.94%)
Mutual labels:  network

yolov3-network-slimming

LICENSE

Learning Efficient Convolutional Networks Through Network Slimming (ICCV 2017)应用在yolov3和yolov2上

环境

pytorch 0.41

window 10

如何使用

1.对原始weights文件进行稀疏化训练

python sparsity_train.py -sr --s 0.0001 --image_folder coco.data --cfg yolov3.cfg --weights yolov3.weights

2.剪枝

python prune.py --cfg yolov3.cfg --weights checkpoints/yolov3_sparsity_100.weights --percent 0.3

3.对剪枝后的weights进行微调

python sparsity_train.py --image_folder coco.data --cfg prune_yolov3.cfg --weights prune_yolov3.weights

关于new_prune.py

new_prune更新了算法,现在可以确保不会有某一层被减为0的情况发生,参考RETHINKING THE SMALLER-NORM-LESSINFORMATIVE ASSUMPTION IN CHANNEL PRUNING OF CONVOLUTION LAYERS(ICLR 2018)对剪枝后bn层β系数进行了保留

待完成

coco测试

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].