34 dépôts
Techniques for preparing raw image data for deep learning model consumption.
Distinct from Data Preprocessing for Modeling: Focuses on the specific domain of preparing image tensors for neural networks, which is narrower than general data preprocessing
Explore 34 awesome GitHub repositories matching artificial intelligence & ml · Image Data Preprocessing. Refine with filters or upvote what's useful.
LivePortrait is a deep learning framework for portrait animation that transfers facial expressions from a driving video to a static image. It functions as an AI motion retargeting tool, mapping movements between different identities while preserving the unique features of the source portrait. The system includes specialized capabilities for cross-species portrait animation, adapting human-centric models to non-human subjects and animals. It also features a motion template generator that converts driving videos into portable files to accelerate inference and protect the identity of the origina
Applies temporal and spatial preprocessing to video sequences to prepare them for motion extraction.
FaceNet is a facial recognition framework designed to transform facial images into high-dimensional numerical embeddings for identity verification and recognition. It provides a deep learning face embedder that maps facial features into a Euclidean space where distance corresponds to facial similarity. The system includes tools for both supervised and unsupervised identity management. It features a face identity classifier for categorizing images into known identity classes and an unsupervised clustering tool to group similar facial embeddings together without predefined labels. The framewor
Standardizes facial images through landmark detection and alignment for better model performance.
This project is a collection of supervised and unsupervised machine learning algorithms implemented from scratch using Python. It serves as an educational resource for studying model training, parameter optimization, and the implementation of core predictive models. The library provides a variety of supervised learning tools, including linear and logistic regression, decision trees, and support vector machines. It also features unsupervised learning capabilities for discovering patterns in unlabeled datasets through clustering algorithms. Broad capability areas include ensemble learning thro
Converts binary image grids into one-dimensional vectors for compatibility with classification algorithms.
MMSegmentation is an open-source semantic segmentation toolbox built on PyTorch that provides a modular, configurable framework for building, training, evaluating, and deploying segmentation models. At its core, it offers a config-driven pipeline that assembles training, evaluation, and inference workflows by parsing hierarchical configuration files, with a modular component registry that enables plug-and-play composition of neural network modules, optimizers, datasets, and metrics. The framework supports the full model lifecycle through a unified runner interface that controls training, testi
Configures pixel normalization, padding, and color channel conversion for input images and segmentation maps.
NSFW detection on the client-side via TensorFlow.js
Converts raw image data into normalised tensor inputs with resizing and channel reordering for the neural network.
mmagic is a multimodal training pipeline and framework for generative AI, focusing on visual synthesis and restoration. It provides the infrastructure to build and train models for tasks such as text-to-image and text-to-video generation, 3D-aware content synthesis, and high-fidelity image translation using diffusion models and generative adversarial networks. The project distinguishes itself through specialized capabilities for generative model personalization, including techniques for fine-tuning subjects and styles. It also supports advanced visual manipulations such as latent space interp
Provides essential preprocessing utilities to normalize pixel values and apply padding to image tensors for model inputs.
Anti-Anti-Spider is an automated web scraping toolkit and CAPTCHA bypass framework. It uses convolutional neural networks to recognize characters and digits in image-based security challenges, enabling programmatic access to protected web content. The project functions as an image recognition model trainer, providing a workflow to preprocess labeled image datasets and train custom neural networks. Users can configure model architectures and hyperparameters to align the recognition system with the visual style of specific target websites. The toolkit covers capabilities for image data preproc
Prepares image datasets for deep learning by resizing and applying label-based naming conventions.
This project is a machine learning educational resource and implementation guide for Python. It provides a collection of executable code and notebooks that demonstrate predictive modeling, data analysis workflows, and the implementation of various machine learning algorithms. The repository features practical examples of classification, regression, and clustering tasks using Scikit-Learn, alongside tutorials for building and training deep learning architectures with TensorFlow. These include implementations of convolutional and recurrent networks. The content covers a broad range of capabili
Implements preprocessing for image files, including channel management and tensor preparation.
mmcv is a foundation library for computer vision based on PyTorch. It provides a comprehensive system for constructing convolutional neural networks, a toolkit for image and video preprocessing, and a collection of high-performance deep learning vision operators. The project is distinguished by its hardware-accelerated kernels for complex operations such as deformable convolutions and region pooling. It features a configuration-driven framework that allows for the dynamic instantiation of network layers and the registration of custom modules without modifying code. The library covers a broad
Provides a toolkit for preparing raw image and video data for deep learning model consumption.
LatentSync est un générateur de vidéo piloté par l'audio et un modèle de synchronisation labiale par diffusion latente conçu pour synchroniser les mouvements des lèvres d'un locuteur dans une vidéo avec une piste audio cible. Il fournit un framework d'entraînement de synchronisation labiale pour développer des réseaux de synchronisation sur des jeux de données vidéo et audio personnalisés. Le système utilise un pipeline de prétraitement vidéo pour nettoyer, segmenter et aligner les données faciales. Il inclut un outil d'évaluation de synchronisation visuelle qui calcule des scores de confiance pour mesurer la précision de l'alignement audio et visuel dans les vidéos générées. Le projet couvre des capacités pour le développement de réseaux de synchronisation personnalisés, la gestion de la configuration d'entraînement pour la mémoire matérielle et la résolution, ainsi que l'évaluation de vidéo synthétique.
Aligns and crops video frames to focus on the mouth region for precise synchronization training.
Ce projet est une collection de supports de cours sur le deep learning avec PyTorch, composée de projets pratiques et d'exercices de programmation. Il se concentre sur l'implémentation d'architectures de réseaux de neurones et l'entraînement de modèles pour résoudre des problèmes de données complexes. Le dépôt inclut une suite de projets de vision par ordinateur pour construire des classificateurs d'images, des auto-encodeurs et des applications de transfert de style. Il propose un laboratoire sur les réseaux antagonistes génératifs (GAN) pour créer des images synthétiques et des implémentations spécifiques pour le transfert d'apprentissage afin d'adapter des poids pré-entraînés à de nouvelles tâches. La base de code couvre l'analyse de données séquentielles pour le traitement du langage naturel (NLP) en utilisant des réseaux de neurones récurrents et des plongements de mots (word embeddings). Des capacités supplémentaires incluent le prétraitement des données d'image, l'évaluation des performances des modèles et le déploiement de modèles entraînés sur une infrastructure cloud. Les supports sont fournis sous forme de série de Jupyter Notebooks.
Implements image loading and augmentation techniques to prepare raw visual data for deep learning.
Ce projet est une ressource pédagogique complète sur le machine learning, présentée sous forme d'une série de tutoriels dans des Jupyter Notebooks interactifs. Il propose des implémentations pratiques en Python pour l'ensemble du cycle de vie du machine learning, couvrant l'apprentissage supervisé et non supervisé, le deep learning et l'apprentissage par renforcement. La ressource se distingue par des guides d'implémentation détaillés pour des architectures complexes, notamment les transformers, les réseaux antagonistes génératifs (GAN) et les réseaux de neurones convolutifs. Elle propose également des cours spécialisés pour développer des agents d'apprentissage par renforcement utilisant le Q-learning et les Deep Q-Networks dans des environnements simulés. Le contenu couvre un large spectre de capacités en data science, incluant les pipelines d'ingénierie de données, l'encodage de caractéristiques et la réduction de dimensionnalité. Il fournit un matériel étendu sur l'évaluation des modèles via la validation croisée et des métriques de diagnostic, ainsi que des sujets avancés comme le traitement du langage naturel (NLP), l'analyse de sentiment et l'IA générative. L'ensemble du cursus est conçu pour une exécution interactive dans des Jupyter Notebooks, combinant code exécutable, texte riche et visualisations.
Provides techniques for preparing raw image data and applying augmentations for deep learning model consumption.
Augmentor est une bibliothèque et un framework Python d'augmentation d'images conçus pour étendre les jeux de données de machine learning. Il fonctionne comme un outil de prétraitement qui génère des variations d'images synthétiques pour accroître la diversité des données et comme un streamer de données d'entraînement qui alimente les boucles de réseaux de neurones avec des images et étiquettes augmentées sans nécessiter de stockage disque intermédiaire. Le framework maintient l'alignement spatial entre les images et leurs masques correspondants, ce qui est requis pour l'entraînement à la segmentation sémantique. Il prend en charge diverses transformations géométriques et au niveau des pixels, incluant les distorsions élastiques, les changements de perspective via inclinaison et déformation, la rotation, le cisaillement et l'effacement aléatoire de régions. Le système inclut des capacités de stratégies de traitement par classe pour remédier au déséquilibre des données et utilise le multi-threading pour accélérer la génération parallèle de jeux de données augmentés. Il fournit également des utilitaires pour nettoyer et standardiser les fichiers images bruts lors de l'étape de prétraitement.
Provides utilities to clean and standardize raw image files for use in machine learning processing pipelines.
This is an image segmentation framework and masking toolkit for constructing binary and multi-class neural network architectures. It serves as a deep learning encoder wrapper that integrates pre-trained convolutional neural network architectures into semantic segmentation models. The library enables the use of pre-trained backbones to isolate complex patterns and leverages transfer learning to accelerate training. It provides a collection of overlap-based loss functions and precision metrics specifically designed to evaluate and refine the accuracy of image masks. The toolkit covers the full
Prepares raw image data to ensure compatibility between data sources and model encoders.
mmocr est un framework de reconnaissance optique de caractères (OCR) basé sur PyTorch conçu pour entraîner et déployer des modèles de détection de texte, de reconnaissance et d'extraction d'informations clés. Il sert de boîte à outils complète pour la détection et la reconnaissance de texte dans les scènes, fournissant des bibliothèques spécialisées pour localiser les régions de texte et convertir le texte visuel en chaînes encodées par machine. Le projet se distingue par un framework de recherche pour l'extraction d'informations clés et des capacités avancées de repérage de texte. Celles-ci incluent le repérage basé sur des points utilisant des transformers et l'utilisation de courbes de Bezier paramétrées pour identifier et transcrire du texte de forme arbitraire. Le framework couvre une large surface de capacités de vision par ordinateur, notamment la gestion de pipeline de données pour augmenter et standardiser divers jeux de données OCR, l'entraînement de modèles avec mise à l'échelle distribuée et l'évaluation des performances utilisant des métriques OCR standard. Il fournit également des utilitaires pour la manipulation de polygones géométriques et la visualisation des résultats pour auditer les prédictions par rapport aux annotations de vérité terrain. Le système est implémenté en Python et prend en charge l'installation via l'empaquetage d'environnement Docker.
Includes essential preprocessing steps like image resizing, polygon rotation, and dataset cleaning to prepare data for OCR models.
This project is a comprehensive instructional resource and course for building neural networks using PyTorch. It covers the fundamental building blocks of deep learning, including tensor manipulation, automatic differentiation, and the construction of modular neural network components. The repository serves as a technical guide for several specialized domains. It provides implementation details for computer vision tasks such as image classification, object detection, and semantic segmentation, as well as natural language processing workflows involving transformers, recurrent networks, and gen
Provides techniques for resizing and normalizing raw images into tensors for deep learning consumption.
img2dataset is a high-performance image dataset pipeline and preprocessing tool designed to download and process millions of images from URLs for machine learning training. It functions as a distributed image downloader and cloud storage data exporter, moving large visual datasets from web sources directly into structured formats. The system prioritizes high-throughput data acquisition by distributing workloads across multiple CPU cores and machines. It integrates directly with remote cloud storage buckets and employs a manifest-based tracking system to resume interrupted downloads without re
Includes a preprocessing suite for resizing, cropping, and filtering images to ensure consistent quality for model training.
This project is a collection of foundational machine learning algorithms and tools implemented from scratch in Python. It serves as a library of core implementations for regression, classification, and clustering models, designed to demonstrate the underlying mathematical structures of these algorithms without relying on high-level machine learning frameworks. The project focuses on the manual implementation of algorithmic logic, including neural networks with forward propagation and weight updates, as well as various supervised and unsupervised learning models. It utilizes NumPy for vectoriz
Provides functionality to normalize pixel values and resize image dimensions for model consumption.
sam-hq est une collection de modèles de fondation de vision pré-entraînés et d'adaptateurs conçus pour la segmentation d'images de haute qualité, l'extraction de caractéristiques multimodales et l'estimation de profondeur. Il fournit un modèle de vision zero-shot capable d'effectuer la segmentation et la classification à travers divers domaines sans nécessiter d'entraînement spécifique à la tâche. Le projet propose un outil de segmentation d'images de haute qualité basé sur le Segment Anything Model qui génère des masques précis à partir de prompts spatiaux. Il inclut un extracteur de caractéristiques multimodal pour générer des embeddings vectoriels de haute dimension à partir d'entrées image et texte, ainsi qu'un outil convolutif pour prédire la distance ou la hauteur de la canopée à partir de données visuelles. Le framework couvre un large éventail de capacités en vision par ordinateur, incluant la classification d'images, l'extraction de caractéristiques multi-résolution et le prétraitement d'images. Il supporte l'adaptation au domaine par fine-tuning sur des jeux de données personnalisés pour des applications spécialisées telles que l'imagerie médicale et la télédétection. Le décodeur de masque peut être converti en un format ouvert pour l'exécution dans des environnements avec un runtime standard.
Scales and normalizes raw image data to ensure compatibility with model input requirements.
ExecuTorch is a lightweight C++ runtime for deploying PyTorch models on mobile, embedded, and edge hardware. It provides an ahead-of-time compilation pipeline that exports, quantizes, and lowers model graphs into compact serialized programs, then executes them through a minimal runtime with hardware acceleration and on-device large language model inference capabilities. The project distinguishes itself through a hardware accelerator delegate system that partitions model subgraphs and offloads computation to specialized backends including NPUs, GPUs, and DSPs from Apple, Arm, Intel, MediaTek,
ExecuTorch keeps platform-dependent image work like decoding, resizing, and cropping in the application layer before passing pixels to the exported model.