Brings classic computer vision into PyTorch as differentiable, GPU-accelerated tensor operators — filters, geometric transforms, feature matching, camera calibration — so each step lives inside autograd and trains end-to-end with neural networks.
Bundles hundreds of pretrained image backbones — ResNet, EfficientNet, ViT, ConvNeXt, Swin and more — behind one consistent API for classification and feature extraction, with training and inference scripts that reproduce published ImageNet results.
Modular implementations of object detection, instance/semantic/panoptic segmentation and related vision models for research and deployment. Offers a large model zoo, export to TorchScript/Caffe2, and PyTorch-native optimizations for faster training and extensibility.
PyTorch object detector built for shipping: train on your own data, then export to ONNX, CoreML, TFLite, or TensorRT with one command. Comes in five sizes (n/s/m/l/x) and adds instance-segmentation and classification heads beyond bounding-box detection.
Library for benchmarking, developing, and deploying deep-learning visual anomaly-detection models — includes ready-to-use model implementations (PatchCore, DINO-based), experiment/HPO tooling, OpenVINO export for edge inference, and a low-code Studio for deployment.
Offline desktop OCR for Windows and Linux that extracts text from screenshots, image batches, and scanned PDFs without requiring a network connection. Bundles multilingual offline engines (PaddleOCR / RapidOCR), supports ignore-regions, searchable PDF output, CLI and HTTP interfaces for automation and integration.
Canonical ILSVRC ImageNet-1k for 1,000-way image classification — provides roughly 1.2M labeled images (train/val/test) packaged as optimized Parquet for easy loading with Hugging Face Datasets, Dask, and Polars. Verify licensing and distribution constraints before use.
Runs pretrained diffusion models for image, video, and audio generation through composable pipelines. It separates pipelines, schedulers, models, adapters, and memory optimizations so teams can prototype quickly without locking into one model family.
Enlarges and enhances low-resolution images using AI models (Real-ESRGAN) through a cross-platform desktop app. Runs on a local NCNN/Vulkan backend (requires a Vulkan-compatible GPU), offers an Electron GUI plus a CLI backend (upscayl-ncnn), and supports custom models for different image types.
Runs a self-hosted web server and React-based UI to generate, edit, and manage images using Stable Diffusion, SDXL and other foundation models. Includes a model manager, unified canvas, node-based workflows, gallery, and support for ckpt, diffusers and some gguf checkpoints.
Browser-based control panel for running Stable Diffusion locally, built on Gradio. Bundles txt2img, img2img, inpainting, outpainting, and upscalers (ESRGAN, GFPGAN, CodeFormer), plus an extension ecosystem and support for NVIDIA, AMD, and Intel GPUs.
Turns text prompts into images through latent diffusion, from local-ready releases to professional SD 3.5 models. Its impact comes from deployability: self-hosting, API access, and community tooling made image generation broadly hackable.