Alan's repositories
DMAE
[CVPR 2023] This repository includes the official implementation our paper "Masked Autoencoders Enable Efficient Knowledge Distillers"
FasterViT
Official PyTorch implementation of FasterViT: Fast Vision Transformers with Hierarchical Attention
deit
Official DeiT repository
FastestDet
:zap: A newly designed ultra lightweight anchor free target detection algorithm, weight only 250K parameters, reduces the time consumption by 10% compared with yolo-fastest, and the post-processing is simpler
External-Attention-pytorch
🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐
difflogic
A Library for Differentiable Logic Gate Networks
ISTDU-Net
code of article ISTDU-Net:Infrared Small-Target Detection U-Net
tiny-training
On-Device Training Under 256KB Memory [NeurIPS'22]
detectron2
Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
QueryDet-PyTorch
[CVPR 2022 Oral] QueryDet: Cascaded Sparse Query for Accelerating High-Resolution Small Object Detection
multispectral-object-detection
Multispectral Object Detection with Yolov5 and Transformer
pytorch-loss
label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful
yolov5-gradcam
Visualizing Yolov5's layers using GradCam
yoloret
Implementation for the paper 'YOLO-ReT: Towards High Accuracy Real-time Object Detection on Edge GPUs'
flash-attention
Fast and memory-efficient exact attention
FGD
Focal and Global Knowledge Distillation for Detectors (CVPR 2022)
ZZUClock
郑大健康状况上报平台脚本
LD
Localization Distillation for Dense Object Detection (CVPR 2022)
labelme
Image Polygonal Annotation with Python (polygon, rectangle, circle, line, point and image-level flag annotation).
torchdistill
A coding-free framework built on PyTorch for reproducible deep learning studies. 🏆20 knowledge distillation methods presented at CVPR, ICLR, ECCV, NeurIPS, ICCV, etc are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy and benchmark.
LSHFM.detection
Object detection of Distilling Knowledge by Mimicking Features.
vovnet-detectron2
VoVNet backbone networks for detectron2, in CVPR 2020
prepare_detection_dataset
convert dataset to coco/voc format
Yolo_Detectron2
Implementation of Yolo using Facebook's Detectron2 (https://github.com/facebookresearch/detectron2) Framework with Quantization support based on AQD: Towards Accurate Quantized Object Detection (https://github.com/aim-uofa/model-quantization)
Yolo-Fastest
:zap: Based on yolo's ultra-lightweight universal target detection algorithm, the calculation amount is only 250mflops, the ncnn model size is only 666kb, the Raspberry Pi 3b can run up to 15fps+, and the mobile terminal can run up to 178fps+