Libraries & frameworks
vision
pytorch/vision
PyTorch's computer vision library with datasets, transforms, and pretrained models.
Topics
Related projects
datasets22.1kA Python library for one-line loading, streaming, and efficient preprocessing of large public datasets for model training and evaluation.
dlib14.5kA modern C++ toolkit of machine learning and vision algorithms with Python bindings for real-world applications.
kornia12.3kA differentiable computer vision library built on PyTorch for image processing, geometric vision, and augmentation pipelines.
ludwig11.8kA declarative PyTorch framework for training, fine-tuning, and serving LLMs and multimodal models from YAML configs.
autogluon10.8kAn AutoML library that delivers strong predictive performance on tabular and time-series data in just a few lines.
spotlight1.3kSpotlight is a Python library that turns dataframes into interactive browser views for exploring images, audio, text and other unstructured data, helping ML teams find clusters and data issues.