awesome-repositories.com
Blog
MCP
awesome-repositories.com

Discover the best open-source repositories with AI-powered search.

ExploreCurated searchesOpen-source alternativesSelf-hosted softwareBlogSitemap
ProjectMCP serverAboutHow we rankPress
LegalPrivacyTerms
© 2026 Bringes Technology SRL·VAT RO45896025·hello@awesome-repositories.com
Back to tommyzihao/train_custom_dataset

Projects sharing features with Train Custom Dataset

30 open-source projects similar to tommyzihao/train_custom_dataset, ranked by shared indexed features. Tags may describe platforms or build tools rather than the same primary purpose. Check each project’s use case, license, and deployment requirements before treating it as a replacement.

  • pair-code/litPAIR-code avatar

    PAIR-code/lit

    3,636View on GitHub↗

    Lit is a machine learning interpretability framework and model debugging tool designed to analyze model behavior and performance. It serves as an interpretability dashboard for large language models and a general performance analyzer for text, image, and tabular datasets. The project distinguishes itself through a comprehensive suite of interpretability tools, including salience map generation for feature attribution, the creation of synthetic and counterfactual examples to test robustness, and the projection of high-dimensional embeddings into visual spaces via UMAP or PCA. It further enable

    TypeScriptmachine-learningnatural-language-processingvisualization
    View on GitHub↗3,636
  • fastai/course-v3fastai avatar

    fastai/course-v3

    4,914View on GitHub↗

    This repository is a comprehensive educational program and deep learning framework designed to teach practical deep learning using PyTorch through notebooks and code examples. It serves as a high-level library for building, training, and deploying neural networks, acting as a model training orchestrator that coordinates PyTorch models, optimizers, and loss functions. The project provides specialized toolkits for computer vision, natural language processing, and tabular data preprocessing. It distinguishes itself through advanced training controls such as discriminative learning rates, a two-w

    Jupyter Notebookdata-sciencedeep-learningfastai
    View on GitHub↗4,914
  • humphd/have-fun-with-machine-learninghumphd avatar

    humphd/have-fun-with-machine-learning

    5,110View on GitHub↗

    This project is a neural network image classifier and a set of tools for building and training convolutional neural networks to recognize and categorize images. It serves as a machine learning educational guide, providing a practical resource for learning neural network fundamentals through an onboarding process. The system includes a dedicated workflow for pretrained model fine-tuning, allowing existing network weights to be adapted to new image categories. This is supported by a transfer learning pipeline that replaces final classification layers and adjusts weights through targeted retrain

    Pythoncaffeimage-classificationmachine-learning
    View on GitHub↗5,110

AI search

Explore more awesome repositories

Describe what you need in plain English — the AI ranks thousands of curated open-source projects by relevance.

Find more with AI search
  • sjwhitworth/golearnsjwhitworth avatar

    sjwhitworth/golearn

    9,438View on GitHub↗

    GoLearn is a machine learning library for the Go programming language. It provides a supervised learning framework and a toolkit for building, training, and evaluating predictive models through a standardized interface. The project implements a data frame system that loads CSV files into structured grids for matrix operations. It includes a preprocessing library for discretizing continuous variables and a model evaluation toolkit that utilizes confusion matrices and cross-validation to measure precision and recall. The library covers data engineering and management, including the ability to

    Go
    View on GitHub↗9,438
  • nyandwi/machine_learning_completeNyandwi avatar

    Nyandwi/machine_learning_complete

    4,983View on GitHub↗

    This is an interactive notebook-based course that teaches machine learning from Python fundamentals through deep learning and natural language processing. It uses real datasets and multiple frameworks within a structured, hands-on curriculum that combines concise explanations with executable code cells, built-in datasets, and embedded exercise checkpoints. Learning progresses through data preparation and exploration, classical machine learning workflows, computer vision with convolutional neural networks, and natural language processing with deep learning, all delivered as a cohesive progressi

    Jupyter Notebookcomputer-visiondata-analysisdata-science
    View on GitHub↗4,983
  • datawhalechina/thorough-pytorchdatawhalechina avatar

    datawhalechina/thorough-pytorch

    3,684View on GitHub↗

    This project is an educational resource and comprehensive guide for implementing and deploying deep learning models using the PyTorch framework. It provides a structured learning curriculum consisting of tutorials and notebooks that cover neural network architectures, data pipelines, and model optimization across multiple AI domains. The curriculum includes practical implementation guides for building convolutional networks, transformers, and recurrent models. It specifically focuses on workflows for computer vision, including image classification, object detection, and segmentation, as well

    Jupyter Notebookdeep-learningmachine-learningpython
    View on GitHub↗3,684
  • open-edge-platform/anomalibopen-edge-platform avatar

    open-edge-platform/anomalib

    5,871View on GitHub↗

    Anomalib is a PyTorch-based library for visual anomaly detection, offering a modular framework, a comprehensive model zoo, and a benchmarking suite designed for industrial defect detection. It provides a wide range of algorithms—including generative, discriminative, teacher-student, and vision-language approaches—that support unsupervised, few-shot, and zero-shot settings. The library enables deployment through model export to ONNX and OpenVINO for edge devices, and includes a no-code web application for training and inference. It also features a command-line interface for orchestrating multi

    Pythonanomaly-detectionanomaly-localizationanomaly-segmentation
    View on GitHub↗5,871
  • dmlc/gluon-cvdmlc avatar

    dmlc/gluon-cv

    5,922View on GitHub↗

    Gluon-CV is an MXNet computer vision library that provides a comprehensive collection of pre-implemented vision architectures and training pipelines. It serves as a deep learning research toolkit and a model zoo containing state-of-the-art pre-trained weights for image and video analysis. The project includes a specialized human pose estimation library and a model compression toolkit. These tools allow for the pruning and quantization of deep learning models to increase inference speed and facilitate deployment on constrained edge hardware. The library covers a broad range of vision capabili

    Pythonaction-recognitioncomputer-visiondeep-learning
    View on GitHub↗5,922
  • mrdbourke/zero-to-mastery-mlmrdbourke avatar

    mrdbourke/zero-to-mastery-ml

    5,839View on GitHub↗

    This project is a machine learning educational curriculum and learning platform delivered through interactive Jupyter Notebooks. It serves as a comprehensive guide for mastering the Python data science toolkit, providing structured tutorials for numerical computing, tabular data manipulation, and statistical visualization. The curriculum includes specific implementation guides for Scikit-Learn and a practical course on TensorFlow for constructing, training, and deploying neural networks and computer vision models. It covers the end-to-end process of building predictive models, from initial pr

    Jupyter Notebookdata-sciencedeep-learningmachine-learning
    View on GitHub↗5,839
  • open-mmlab/mmdetection3dopen-mmlab avatar

    open-mmlab/mmdetection3d

    6,273View on GitHub↗

    MMDetection3D is an open-source toolbox for 3D perception, providing a unified framework for detecting and segmenting objects in three-dimensional environments. It supports a range of core tasks including monocular 3D object detection from single camera images, LiDAR-based 3D object detection from raw point clouds, and multi-modal fusion that combines camera images with LiDAR data. The toolbox also covers point cloud semantic segmentation, assigning class labels to every point in a scan for scene understanding. The project distinguishes itself through a config-driven pipeline that orchestrate

    Python3d-object-detectionobject-detectionpoint-cloud
    View on GitHub↗6,273
  • fastai/course22fastai avatar

    fastai/course22

    3,398View on GitHub↗

    This is a structured deep learning curriculum for programmers, delivered as a collection of Jupyter notebooks. It teaches the fundamentals of training neural networks for computer vision, natural language processing, tabular data analysis, and collaborative filtering using PyTorch and the fastai library. The course is designed to be hands-on, guiding learners from building a training loop from scratch to fine-tuning pretrained models for a variety of practical tasks. The curriculum distinguishes itself by covering the full lifecycle of a deep learning project, from data preparation and augmen

    Jupyter Notebookdeep-learningfastaijupyter-notebooks
    View on GitHub↗3,398
  • thu-mig/yolov10THU-MIG avatar

    THU-MIG/yolov10

    11,316View on GitHub↗

    YOLOv10 is a PyTorch computer vision library and real-time vision framework designed for locating and identifying multiple objects in images and video streams. It functions as an end-to-end object detector that optimizes for high-speed deployment and detection precision. The project is distinguished by an NMS-free detection architecture that predicts a single bounding box per object, eliminating the need for non-maximum suppression post-processing to reduce inference latency. It further optimizes for edge hardware through scalable weights and a quantization-friendly structure that facilitates

    Python
    View on GitHub↗11,316
  • paddlepaddle/paddleclasPaddlePaddle avatar

    PaddlePaddle/PaddleClas

    5,816View on GitHub↗

    PaddleClas is a toolkit for image classification and recognition built on PaddlePaddle. It provides a suite of tools for training deep learning models and a framework for implementing visual search and retrieval systems. The project includes a computer vision model optimization suite and tools for cross-platform deployment. It enables the export of trained models to servers, mobile devices, and edge hardware to achieve high-performance inference across different programming languages. The toolkit covers model compression and optimization through pruning, quantization, and knowledge distillat

    Pythonautoaugmentcutmixdeit
    View on GitHub↗5,816
  • olafenwamoses/imageaiOlafenwaMoses avatar

    OlafenwaMoses/ImageAI

    8,867View on GitHub↗

    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

    Pythonai-practice-recommendationsalgorithmartificial-intelligence
    View on GitHub↗8,867
  • microsoft/computervision-recipesmicrosoft avatar

    microsoft/computervision-recipes

    9,866View on GitHub↗

    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

    Jupyter Notebookartificial-intelligenceazurecomputer-vision
    View on GitHub↗9,866
  • promptslab/promptifypromptslab avatar

    promptslab/Promptify

    4,616View on GitHub↗

    Promptify is a suite of tools designed for model evaluation, prompt management, token cost tracking, structured extraction, and unified API gateway access. It provides a standardized interface to manage requests and responses across multiple large language model providers. The project features a prompt management platform for engineering and versioning prompts with structured output validation. It includes a dedicated evaluation framework to measure model performance using precision, recall, and f1 scores against labeled datasets, alongside a token cost tracker to monitor the financial expens

    Python
    View on GitHub↗4,616
  • yahoo/open_nsfwyahoo avatar

    yahoo/open_nsfw

    6,011View on GitHub↗

    Open NSFW is a deep learning image classifier and content filter designed to identify explicit or unsafe visual content. It uses a neural network to analyze images and generate probability scores that determine whether media is not safe for work. The system employs a Caffe-based model architecture to perform automated visual feature extraction and image classification. It supports the use of pre-trained weights and allows for fine-tuning the network with custom datasets to adjust detection accuracy. The project covers end-to-end image moderation workflows, including automated content scannin

    Python
    View on GitHub↗6,011
  • biolab/orange3biolab avatar

    biolab/orange3

    5,635View on GitHub↗

    Orange3 is a visual data mining platform that provides an interactive canvas for building data analysis workflows without writing code. At its core, it offers a widget-based visual programming environment where users connect configurable components to perform data preprocessing, machine learning model training, statistical evaluation, and interactive visualization. The platform is built on NumPy-backed data tables with domain descriptors that define variable names, types, and roles, and includes a lazy SQL query proxy for working with database tables without loading all data into memory. The

    Python
    View on GitHub↗5,635
  • mrdbourke/tensorflow-deep-learningmrdbourke avatar

    mrdbourke/tensorflow-deep-learning

    5,914View on GitHub↗

    This is a comprehensive deep learning course delivered entirely through Jupyter Notebooks, designed to teach neural network construction using TensorFlow 2.x. The curriculum follows a sequential-model-first pedagogy, introducing the Sequential API before moving to functional and subclassing approaches, and covers the full spectrum of model building from regression and classification through convolutional neural networks, natural language processing, and time series forecasting. The course is structured around a checkpoint-based training workflow that saves the best model weights during traini

    Jupyter Notebook
    View on GitHub↗5,914
  • microsoft/nlp-recipesmicrosoft avatar

    microsoft/nlp-recipes

    6,436View on GitHub↗

    nlp-recipes is a collection of implementation guides and reference templates for applying natural language processing techniques to real-world tasks. It provides standardized workflows and code examples for developing NLP pipelines, from dataset preparation and model training to performance evaluation. The project focuses on the practical application of transformer-based models, offering patterns for fine-tuning pretrained architectures for tasks such as text classification, named entity recognition, and question answering. It also includes a toolkit for model interpretability, allowing users

    Python
    View on GitHub↗6,436
  • open-mmlab/mmpretrainopen-mmlab avatar

    open-mmlab/mmpretrain

    3,842View on GitHub↗

    mmpretrain is a modular PyTorch computer vision framework designed for developing, training, and benchmarking deep learning architectures. It serves as a comprehensive toolkit for vision tasks, providing a specialized platform for multimodal machine learning and self-supervised learning. The project features a computer vision model zoo containing architectural definitions and pre-trained weights for backbones such as ViT, ConvNeXt, and Swin Transformer. It distinguishes itself through a dedicated self-supervised learning toolkit that implements algorithms like MAE and DINO to train models wit

    Pythonbeitclipconstrastive-learning
    View on GitHub↗3,842
  • paddlepaddle/paddlexPaddlePaddle avatar

    PaddlePaddle/PaddleX

    6,163View on GitHub↗

    PaddleX is a PaddlePaddle-based framework for building, deploying, and fine-tuning AI model pipelines, with pre-built support for computer vision, OCR, document analysis, and time series tasks. It offers a toolkit of ready-to-use pipelines for image classification, object detection, segmentation, and pose estimation, alongside an end-to-end OCR document analysis pipeline that extracts text, tables, formulas, and layout information. The platform also includes a dedicated time series forecasting pipeline for analyzing historical data to detect anomalies, classify patterns, and predict future val

    Pythonai-pipelinesclassificationdeployment
    View on GitHub↗6,163
  • zalandoresearch/fashion-mnistzalandoresearch avatar

    zalandoresearch/fashion-mnist

    12,754View on GitHub↗

    This project is a computer vision benchmark and image classification dataset used to measure and compare the accuracy of machine learning models. It provides a standardized collection of labeled fashion product images and training data formatted to be compatible with the MNIST dataset structure. The dataset consists of fixed-dimension grayscale images and label-based category mappings, stored in a binary format. It includes pre-split training and testing sets and a static distribution to ensure consistent cross-model benchmarking. The repository supports image classification benchmarking and

    Pythonbenchmarkcomputer-visionconvolutional-neural-networks
    View on GitHub↗12,754
  • akramz/hands-on-machine-learning-with-scikit-learn-keras-and-tensorflowAkramz avatar

    Akramz/Hands-on-Machine-Learning-with-Scikit-Learn-Keras-and-TensorFlow

    1,041View on GitHub↗

    This project serves as an educational and practical resource for mastering machine learning workflows using Python. It provides a comprehensive collection of code examples and exercises designed to guide users through the implementation of predictive systems, ranging from fundamental algorithms to deep learning architectures. The repository distinguishes itself by offering a structured approach to both classical machine learning and neural network training. It covers the full lifecycle of model development, including the orchestration of reusable data transformation pipelines, advanced ensemb

    Jupyter Notebookartificial-intelligencedeep-learningmachine-learning
    View on GitHub↗1,041
  • paddlepaddle/paddledetectionPaddlePaddle avatar

    PaddlePaddle/PaddleDetection

    14,243View on GitHub↗

    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

    Pythonblazefacedeepsortdetr
    View on GitHub↗14,243
  • tingsongyu/pytorch_tutorialTingsongYu avatar

    TingsongYu/PyTorch_Tutorial

    8,018View on GitHub↗

    This project is a comprehensive collection of educational examples and reference implementations for building vision and language models using PyTorch. It serves as a deep learning tutorial covering the end-to-end process of developing neural networks, from initial architecture definition to final production deployment. The repository provides detailed guides on implementing a wide range of domain-specific models, including convolutional neural networks for object detection and segmentation, as well as transformer and recurrent architectures for natural language processing. It emphasizes gene

    Python
    View on GitHub↗8,018
  • districtdatalabs/yellowbrickDistrictDataLabs avatar

    DistrictDataLabs/yellowbrick

    4,398View on GitHub↗

    Yellowbrick is a machine learning visualization library and model diagnostic tool designed to analyze feature importance, target distributions, and model error metrics. It serves as a visual toolkit for diagnosing underfitting and overfitting through the use of validation and learning curves. The project provides specialized suites for evaluating predictive models and unsupervised learning. It enables the determination of optimal cluster counts via elbow methods and silhouette coefficients, and assesses classifier and regressor quality through ROC curves, confusion matrices, and residual plot

    Python
    View on GitHub↗4,398
  • open-mmlab/mmaction2open-mmlab avatar

    open-mmlab/mmaction2

    5,066View on GitHub↗

    mmaction2 is a PyTorch video understanding toolbox designed for training and evaluating deep learning models. It serves as a framework for action recognition, temporal localization, and spatio-temporal action detection, providing specialized tools for both pixel-based video analysis and skeleton-based action recognition. The project distinguishes itself through a modular architecture featuring registry-based component discovery and hierarchical, config-driven model assembly. It supports multi-modal feature fusion, integrating RGB frames, optical flow, and audio, and includes capabilities for

    Python
    View on GitHub↗5,066
  • deepchecks/deepchecksdeepchecks avatar

    deepchecks/deepchecks

    4,024View on GitHub↗

    Deepchecks is a machine learning model validation framework and MLOps testing library. It serves as an AI data quality suite and performance evaluator designed to verify the integrity and performance of models and datasets from research through production. The project functions as a model monitoring tool for tracking data drift and performance degradation in production environments. It allows for the creation of custom validation suites and utilizes a pluggable check architecture to automate quality checks within continuous integration pipelines. The framework covers a broad range of capabil

    Python
    View on GitHub↗4,024
  • instillai/machine-learning-courseinstillai avatar

    instillai/machine-learning-course

    7,043View on GitHub↗

    This is a comprehensive educational curriculum designed to teach machine learning fundamentals using the Python programming language. It provides a structured course covering the implementation and theory of supervised learning, unsupervised learning, and deep learning. The curriculum is delivered through interactive notebooks that combine executable code with technical tutorials. It includes dedicated guides for building neural network architectures, implementing classification and regression models, and utilizing clustering techniques for pattern discovery in unlabeled data. The materials

    Python
    View on GitHub↗7,043