Example 📓 Jupyter notebooks that demonstrate how to build, train, and deploy machine learning models using 🧠 Amazon SageMaker.

Image augmentation for machine learning experiments.

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

Segmentation models with pretrained backbones. PyTorch.

Data, tools, and documentation of the Fusion 360 Gallery Dataset

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

[T-PAMI-2024] Transformer-Based Visual Segmentation: A Survey

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

[NeurIPS 2024] Depth Anything V2. A More Capable Foundation Model for Monocular Depth Estimation

Depth Pro: Sharp Monocular Metric Depth in Less Than a Second.

🕸️ A Python interface to CGAL's meshing tools

A collection of meshes for canonical domains

Official implementation of DeepLabCut: Markerless pose estimation of user-defined features with deep learning for all animals incl. humans

Grounding Image Matching in 3D with MASt3R

We write your reusable computer vision tools. 💜

Applied Deep Learning Course

Fast and memory-efficient exact attention

Efficient Triton Kernels for LLM Training

Python bindings for COLMAP

2D/3D simplicial mesh generator interface for Python (Triangle, TetGen, gmsh)

Materials for the Learn PyTorch for Deep Learning: Zero to Mastery course.

[ICRA 2022] CaTGrasp: Learning Category-Level Task-Relevant Grasping in Clutter from Simulation

Hydra is a framework for elegantly configuring complex applications

Fast Graphlet Transform

3D plotting and mesh analysis through a streamlined interface for the Visualization Toolkit (VTK)

A Python toolbox for performing gradient-free optimization

The AI Datastore for Schemas, BLOBs, and Predictions. Use with your apps or integrate built-in Human Supervision, Data Workflow, and UI Catalog to get the most value out of your AI Data.

Label Studio is a multi-type data labeling and annotation tool with standardized output format

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