Stars
Provide with pre-build flash-attention package wheels using GitHub Actions
2nd Place Solution for the RSNA 2023 Abdominal Trauma Detection Kaggle Competition
Painter & SegGPT Series: Vision Foundation Models from BAAI
Simple jupyter notebook for 3D reconstruction using kornia and pycolmap
python library for invisible image watermark (blind image watermark)
High-Resolution Image Synthesis with Latent Diffusion Models
Set of methods to ensemble boxes from different object detection models, including implementation of "Weighted boxes fusion (WBF)" method.
SOTA Semantic Segmentation Models in PyTorch
Solution of Kaggle competition: Sartorius - Cell Instance Segmentation
PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
[CVPR 2022] This repository includes the official project for the paper: TransMix: Attend to Mix for Vision Transformers.
Python package for reading, and extracting data from rosbag files and performing any analysis on it.
Official PyTorch Implementation of MotionSqueeze, ECCV 2020
BoxMOT: pluggable SOTA tracking modules for segmentation, object detection and pose estimation models
PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
My 1st place solution at Kaggle Hotel-ID 2021
Unofficial implementation of PatchCore anomaly detection
Vision Transformer for 3D medical image registration (Pytorch)
Flexible Python configuration system. The last one you will ever need.
Reproducing CheXpert paper using the PyTorch library.
Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers
Benchmark inference speed of CNNs with various quantization methods in Pytorch+TensorRT with Jetson Nano/Xavier
Code for our ICASSP 2021 paper: SA-Net: Shuffle Attention for Deep Convolutional Neural Networks
Semi Supervised Learning for Medical Image Segmentation, a collection of literature reviews and code implementations.