UNetFormer: A UNet-like transformer for efficient semantic segmentation of remote sensing urban scene imagery, ISPRS. Also, including other vision transformers and CNNs for satellite, aerial image and UAV image segmentation.

Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models

Extract video features from raw videos using multiple GPUs. We support RAFT flow frames as well as S3D, I3D, R(2+1)D, VGGish, CLIP, and TIMM models.

Simple Implementation of Pix2Seq model for object detection in PyTorch

Encoder-Decoder Cell and Nuclei segmentation models

PyTorch Volume Models for 3D data

A PyTorch-based Python library with UNet architecture and multiple backbones for Image Semantic Segmentation.

💡💡💡awesome compute vision app in gradio

Jittor Image Models is a library for pulling together a wide variety of SOTA deep learning models in the Jittor framework.

A library that includes Keras3 layers, blocks and models with pretrained weights, providing support for transfer learning, feature extraction, and more.

Attribution methods that explain image classification models, implemented in PyTorch, and support batch inputs and GPU.

Simple image classification for custom dataset (pytorch-lightning, timm)

Clean, reproducible, boilerplate-free deep learning project template.

PyTorch Lightning + Hydra. + Timm: A very user-friendly template for rapid and reproducible MLOps with best practices. ⚡🔥⚡

Benchmarks for popular neural network models supported by timm

Deep learning based color transfer between images.

🥈12th place solution on G2Net Detecting Continuous Gravitational Waves🥈

The AI framework for software engineers

Auto torch image models: train and evaluation

This application utilizes the powerful Florence-2 vision-language model from Microsoft to generate comprehensive captions for images. The model is capable of understanding visual content and expressing it in natural language.