Making large AI models cheaper, faster and more accessible

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.

Image-to-Image Translation in PyTorch

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

A list of all named GANs!

Geometric Computer Vision Library for Spatial AI

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

YOLOv10: Real-Time End-to-End Object Detection

A collection of machine learning examples and tutorials.

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

Repo for external large-scale work

😎 Finding duplicate images made easy!

gpt-4o for windows, macos and linux

A collection of loss functions for medical image segmentation

Denoising Diffusion Probabilistic Models

Unofficial implementation of Image Super-Resolution via Iterative Refinement by Pytorch

Tutorial material on the scientific Python ecosystem

View model summaries in PyTorch!

This repository includes the official project of TransUNet, presented in our paper: TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation.

A Python library for creating LaTeX files

Many studies have shown that the performance on deep learning is significantly affected by volume of training data. The MedicalNet project provides a series of 3D-ResNet pre-trained models and rela…

Synchronized Batch Normalization implementation in PyTorch.

[CVPR 2022] Official implementation of the paper "Uformer: A General U-Shaped Transformer for Image Restoration".

Official Pytorch Implementation of: "Asymmetric Loss For Multi-Label Classification"(ICCV, 2021) paper

Automated lung segmentation in CT

This repo contains the code for our paper "A novel focal Tversky loss function and improved Attention U-Net for lesion segmentation" accepted at IEEE ISBI 2019.

Tomographic Reconstruction in Python

Comparison of IQA models in Perceptual Optimization