BillyTheKids-QRR's starred repositories
Awesome-Knowledge-Distillation
Awesome Knowledge-Distillation. 分类整理的知识蒸馏paper(2014-2021)。
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
ConvTransformerTimeSeries
Convolutional Transformer for time series
FGDAE-a-machinery-fault-detection-method
一种用于复杂工况机械故障检测的GNN方法
Complement-Cross-Entropy
Imbalanced Image Classification with Complement Cross Entropy (Pytorch)
HIT-dataset
This is a dataset of inter-shaft bearing based on the vibration signal of rotors and casings, which comes from a aero-engine test with inter-shaft bearing fault. Due to the large size of the data set file, we uploaded the dataset to Google Drive with the link as: https://drive.google.com/drive/folders/1Km1Go4ilB_bI033SBJ7eJ0uCzbqEqbgt?usp=sharing
Loss_ToolBox-PyTorch
PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss
pure-noise
Official implementation for "Pure Noise to the Rescue of Insufficient Data: Improving Imbalanced Classification by Training on Random Noise Images" https://arxiv.org/abs/2112.08810
keras-unet-collection
The Tensorflow, Keras implementation of U-net, V-net, U-net++, UNET 3+, Attention U-net, R2U-net, ResUnet-a, U^2-Net, TransUNET, and Swin-UNET with optional ImageNet-trained backbones.
GraphSmote
Pytorch implementation of paper 'GraphSMOTE: Imbalanced Node Classification on Graphs with Graph Neural Networks' to appear on WSDM2021
easy-few-shot-learning
Ready-to-use code and tutorial notebooks to boost your way into few-shot learning for image classification.
ProtoTransfer
Official code for the paper "Self-Supervised Prototypical Transfer Learning for Few-Shot Classification"
improved-diffusion
Release for Improved Denoising Diffusion Probabilistic Models
stable-diffusion
A latent text-to-image diffusion model