bobo504 Goto Github PK

attention-mechanism-implementation

Self-attention、Non-local、SE、SK、CBAM、DANet

attention-module

Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"

bobo504_github

deep-learning-for-image-processing

deep learning for image processing including classification and object-detection etc.

ecanet

Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks

external-attention-pytorch

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

hardhat10k

imgclsmob

Sandbox for training deep learning networks

mobile-yolov5-pruning-distillation

mobilev2-yolov5s剪枝、蒸馏，支持ncnn，tensorRT部署。ultra-light but better performence！

object-detection-knowledge-distillation

An Object Detection Knowledge Distillation framework powered by pytorch, now having SSD and yolov5.

objectdetection_script

一些关于目标检测的脚本的改进思路代码，详细请看readme.md

pest-yolo

Pest-yolo: A lightweight YOLOv5n-based pest detection algorithm

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

torch-template-for-deep-learning

Pytorch Implementations of large number classical backbone CNNs, data enhancement, torch loss, attention, visualization and some common algorithms.

yolo-gradcam

yolo model with gradcam visual.

yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏

yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

yolov5-distillation-5.0

yolov5 5.0 version distillation || yolov5 5.0版本知识蒸馏，yolov5l >> yolov5s

yolov5-distillation-train-inference

Yolov5 distillation training | Yolov5知识蒸馏训练，支持训练自己的数据

yolov5-distilling

yolov5 knowledge distilling

yolov5-multibackbone-compression

YOLOv5 Series Multi-backbone(TPH-YOLOv5, Ghostnet, ShuffleNetv2, Mobilenetv3Small, EfficientNetLite, PP-LCNet, SwinTransformer YOLO), Module(CBAM, DCN), Pruning (EagleEye, Network Slimming), Quantization (MQBench) and Deployment (TensorRT, ncnn) Compression Tool Box.