Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

Fast image augmentation library and an easy-to-use wrapper around other libraries. Documentation: https://albumentations.ai/docs/ Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

Official repository of FLatten Transformer (ICCV2023)

[ICLR'23 Spotlight] The first successful BERT/MAE-style pretraining on any convolutional network; Pytorch impl. of "Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modeling"

📝Minecraft developer Chinese guide，我的世界开发者中文指南

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Unofficial PyTorch implementation of Masked Autoencoders Are Scalable Vision Learners

Submanifold sparse convolutional networks

MixMIM: Mixed and Masked Image Modeling for Efficient Visual Representation Learning

PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

Official Pytorch Code for "Medical Transformer: Gated Axial-Attention for Medical Image Segmentation" - MICCAI 2021

Code release for ConvNeXt model

✔️李沐 【动手学深度学习】课程学习笔记：使用pycharm编程，基于pytorch框架实现。

医学影像数据集列表 『An Index for Medical Imaging Datasets』

浙江大学胡浩基《机器学习》代码

手写实现李航《统计学习方法》书中全部算法

vit的pytorch实现版本，任务为简单的图片分类

Generate Evangelion title card like a boss 😎

PyTorch implementation of MoCo v3 https//arxiv.org/abs/2104.02057

李宏毅《机器学习》笔记，在线阅读地址：https://datawhalechina.github.io/leeml-notes

Implementation of Axial attention - attending to multi-dimensional data efficiently

a pytorch implementation for MoCo V3

Implementation of SETR model, Original paper: Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers.

Get a large image dataset with minimal effort by grabbing image through the web and generate new ones by image augmentation.

Implementation of UNET, SEGNET and DeepLabV3Plus on MoNuSeg Dataset