Awesome GitHub RepositoriesComputer Vision Training

DeepSpeed is a high-performance library designed to scale deep learning model training and inference across massive clusters of GPUs and compute nodes. It provides a comprehensive suite of tools for distributed training, enabling the execution of models that exceed the memory capacity of single devices through advanced parameter partitioning, pipeline-based model parallelism, and memory-efficient state offloading. The framework distinguishes itself through specialized communication-efficient optimizers and hardware-aware acceleration techniques. By utilizing gradient compression, quantization

This project is a library of pretrained computer vision architectures and backbones for image classification and feature extraction. It serves as a comprehensive model zoo and collection of standardized image encoders, including ResNet, Vision Transformers, and EfficientNet, for use in visual analysis and as backbones for object detection and image segmentation. The library provides a framework for distributed training and evaluation of image models using advanced data augmentation and optimization scripts. It includes a dedicated toolset for converting trained PyTorch vision models into the

This project is a comprehensive library of state-of-the-art neural network architectures designed for image classification and feature extraction. It provides a complete deep learning training framework that supports distributed execution, allowing users to build, train, and fine-tune vision models using optimized schedulers and pre-configured training recipes. The library distinguishes itself through a modular backbone architecture that treats neural networks as decoupled feature extractors, enabling the retrieval of multi-scale outputs for downstream tasks like object detection and segmenta

Swin-Transformer is a deep learning framework designed for training and deploying hierarchical vision transformer models. It serves as a research library and toolkit for computer vision tasks, providing the infrastructure to build models that replace standard convolution operations with sliding window self-attention mechanisms. By utilizing a multi-scale feature hierarchy, the framework enables the processing of visual data at varying resolutions and spatial scales. The project distinguishes itself through its implementation of shifted window partitioning, which facilitates global information

PaddleDetection is an object detection framework designed for the end-to-end development, training, and deployment of computer vision models. It provides a comprehensive library of modular neural network architectures and pipelines that support object detection, instance segmentation, and multi-object tracking tasks. The project distinguishes itself through a configuration-driven approach that decouples model components like backbones and heads, allowing for the flexible assembly of custom vision workflows. It incorporates advanced techniques such as anchor-free detection logic, joint detecti

This project is a machine learning data pipeline designed to automate the collection, curation, and preparation of large-scale image datasets. It functions as an image dataset scraper and computer vision curator, providing the necessary infrastructure to aggregate categorized files from web sources and organize them into structured directories for model development. The system distinguishes itself through a batch-processing architecture that integrates data acquisition with automated integrity validation. By scanning files to remove corrupted or invalid images and applying deterministic parti

This project is a collection of educational resources and implementation frameworks providing deep learning model recipes, code samples, and step-by-step guides for computer vision tasks. It organizes complex workflows into modular recipes and implementation guides to facilitate the building of image and video analysis models. The framework focuses on specialized vision capabilities, including an image similarity framework for fast retrieval and re-ranking, human pose estimation, and video action recognition. It also provides specific tools for crowd density estimation and document image clea

Keras-GAN is a collection of generative adversarial network implementations built with Keras for synthetic data generation and image manipulation. It provides frameworks for image-to-image translation, image inpainting, and neural image super-resolution. The library includes tools for learning disentangled latent space representations to control specific attributes of synthetic outputs. It also features capabilities for image domain translation using paired or unpaired data and the ability to fill corrupted or missing image parts by analyzing surrounding visual context. The project covers ge

ImageAI is a Python computer vision library providing a suite of tools for image classification, object detection, and video analytics. It functions as an integrated framework for locating and labeling objects in static images and video streams, utilizing deep learning models for identification and categorization. The project includes a model training toolkit that allows for the creation of custom classifiers and detectors through scratch training or transfer learning. It features a GPU-accelerated inference engine to increase processing speed for vision tasks and includes specialized utiliti

DAIN is a video frame synthesis engine and AI video upsampling tool designed to increase video playback smoothness. It functions as a computer vision model that synthesizes intermediate frames between existing images to transform low frame rate video into high frame rate content. The system utilizes depth-aware video frame interpolation to predict the motion of pixels between consecutive images. By analyzing spatial depth via depth maps, the tool generates new frames that account for occlusions and overlapping objects to create slow motion effects. The framework incorporates optical flow int

Infinigen is a procedural 3D scene generation framework that creates photorealistic indoor and outdoor environments for computer vision training data. It combines constraint-based object placement, GPU geometry shaders, and ground-truth rendering passes to produce scenes with depth, normals, and segmentation masks alongside final images. The framework distinguishes itself through modular asset composition, a node-graph material system, and physics simulation integration that embeds rigid-body and fluid dynamics directly into the generation pipeline. Procedural rule-based scene composition and

Corenet is a deep learning training framework and computer vision model library designed for developing neural networks across vision, text, and audio modalities. It functions as a distributed training orchestrator for scaling workloads across multiple compute nodes and provides a multimodal data pipeline for processing image, text, and video data. The project includes a model conversion toolkit for transforming weights and architectures between different machine learning frameworks. It also provides tools for optimizing model performance on Apple Silicon and reducing response latency in gene

EasyPR is an automatic license plate recognition system designed to detect vehicle license plates and extract alphanumeric characters from images of Chinese vehicles. It functions as a deep learning OCR tool that converts image regions of license plates into machine-readable text strings. The system includes a specialized detector for identifying vehicle plates within unconstrained environments and complex visual backgrounds. It also provides a synthetic data generator to create artificial image datasets used to train and improve the accuracy of the recognition models. The project covers a m

This project is a collection of TensorFlow 2.x machine learning tutorials and practical code examples. It serves as a deep learning implementation guide for constructing diverse neural network architectures, including convolutional, recurrent, and generative networks. The repository provides templates and examples for several specialized domains, including computer vision for image classification and object detection, natural language processing for text generation and language understanding, and generative AI for synthesizing data using adversarial networks and autoencoders. It also includes

هذا المشروع عبارة عن إطار عمل لمعالجة اللغات الطبيعية يركز على نموذج تدريب مسبق ذاتي الانحدار ومعمم مصمم لتمثيل اللغة غير الخاضع للإشراف. ينفذ نموذج لغة يجمع بين التدريب القائم على التبديل وعمود فقري من نوع Transformer-XL ليعمل كمعالج نصوص طويل السياق. يتميز النظام بقدرته على التعامل مع تسلسلات النصوص التي تتجاوز حدود الطول القياسية من خلال استخدام التكرار على مستوى القطعة والترميز الموضعي النسبي. يقوم بتوسيع نطاق التدريب المسبق عالي الأداء عبر وحدات GPU متعددة وعناقيد TPU باستخدام تطبيقات التدريب الموزع. يغطي الكود المصدري سير عمل التعلم الآلي بالكامل، بما في ذلك تنظيف النصوص وتقسيم الكلمات الفرعية للمعالجة المسبقة للبيانات، بالإضافة إلى الضبط الدقيق الخاص بالمهمة للإجابة على الأسئلة، وفهم القراءة، وتصنيف النصوص. يتضمن أدوات لتحسين المعلمات، وجدولة معدل التعلم، وتقييم احتمالات الإجابة من خلال مقاييس الدقة والاستدعاء. يوفر المشروع تكوينات لإدارة المعلمات الفائقة للنموذج والتدريب المسرع بالأجهزة عبر مضيفين متعددين.

هذا المشروع عبارة عن مجموعة من دفاتر Jupyter التعليمية التي تقدم دروساً حول بناء الشبكات العصبية وعمليات المصفوفات (tensors) باستخدام إطار عمل TensorFlow. يعمل كمستودع تعليمي للتعلم الآلي ودليل تنفيذ لطلاب التعلم العميق. تركز المجموعة على بنيات متقدمة محددة، بما في ذلك الشبكات التلافيفية (CNN) لتصنيف الصور، والشبكات المتبقية (residual networks) مع وصلات التخطي لضمان استقرار التدريب، والمشفرات التلقائية المتغيرة (variational autoencoders) للنمذجة التوليدية وتوليف البيانات. كما تتضمن أدلة لبناء المشفرات التلقائية لإزالة الضوضاء والعميقة لاستخراج الميزات وتقليل الأبعاد. يغطي المستودع نطاقاً أوسع من النمذجة التنبؤية، مع تنفيذ الانحدار الخطي ومتعدد الحدود واللوجستي للتنبؤ بالقيم المستمرة والنتائج الثنائية. المحتوى منظم في دفاتر تفاعلية تسمح للمستخدمين بتنفيذ العمليات الرياضية وتعديل تجارب التعلم الآلي.

Neuraltalk2 هو نظام رؤية للتعلم العميق مصمم للتعليق التلقائي على الصور. مبني باستخدام PyTorch، ويستخدم بنية هجينة تجمع بين مشفر شبكة عصبية تلافيفية (CNN) وفك تشفير شبكة عصبية متكررة (RNN) لتوليد أوصاف نصية من المدخلات المرئية. يتميز المشروع بخط تدريب مسرع بواسطة GPU قادر على توزيع أعباء العمل عبر وحدات معالجة رسومات متعددة من خلال التوزيع متعدد العمليات. يدعم توليد أوصاف لكل من ملفات الصور الثابتة وتيارات الفيديو في الوقت الفعلي. يتضمن إطار العمل قدرات لضبط المشفر، وأخذ عينات نصية عبر البحث الشعاعي (Beam search) مع التحكم في درجة الحرارة، واستخدام مقاييس اللغة القياسية في الصناعة لتقييم دقة وطلاقة التعليق. كما يوفر أدوات لمعالجة البيانات مسبقًا، واستمرارية نقاط فحص النموذج، وتصدير التنبؤات إلى ملفات JSON مهيكلة. يتم توفير التنفيذ كدفتر ملاحظات Jupyter.

Super-Gradients هو إطار عمل للرؤية الحاسوبية لـ PyTorch ومكتبة تدريب مصممة لدورة حياة نماذج الرؤية بالكامل. يعمل كمحسن لنماذج تعلم الآلة ومجموعة أدوات نشر لتدريب وضبط النماذج عبر مهام تصنيف الصور، واكتشاف الكائنات، والتجزئة الدلالية، وتقدير الوضع. يوفر المشروع أدوات محددة لتحسين النماذج، بما في ذلك تقطير المعرفة (Knowledge Distillation) وضغط الدقة الرقمية لتقليل متطلبات الذاكرة والحوسبة. كما يتضمن تنفيذ بنية Yolo-NAS لاكتشاف الكائنات عالي الأداء. يغطي إطار العمل سطح قدرات واسع بما في ذلك التدريب الموزع على GPU، وخطوط أنابيب الرؤية المعيارية، وأتمتة عمليات التدريب عبر تكوينات الوصفات المهيكلة. كما يدير تحميل البيانات، وتعزيز الصور، وتصدير الأوزان المدربة إلى تنسيقات عالمية لمسرعات الأجهزة الإنتاجية.

Co-tracker is a PyTorch point tracking framework and dense point tracking model designed to map the motion of individual pixels throughout a video. It functions as a video pixel tracker that predicts point trajectories and visibility masks across sequences of video frames. The project includes a computer vision training pipeline that utilizes teacher-student knowledge distillation. This allows for the generation of pseudo-labels from unannotated real video data to fine-tune pre-trained models and reduce the gap between synthetic and real data environments. The framework provides capabilities

Neuralangelo is a neural surface reconstruction framework that transforms two-dimensional image sequences and multi-view photography into high-fidelity 3D meshes. It implements a pipeline for training neural radiance fields to represent complex scenes as digital geometry. The project utilizes a signed distance function for surface representation and multi-resolution hash encoding to capture both coarse and fine geometric details. It employs differentiable volume rendering and gradient-based eikonal regularization to ensure the learned distance functions remain physically plausible. The syste