training-pipeline

Neural networks training pipeline based on PyTorch

My repo for training neural nets using pytorch-lightning and hydra

Curriculum training

YOLOv12 Underwater Object Detection is an open-source suite for underwater object detection, built on YOLOv12. It offers an end-to-end pipeline with GPU-accelerated training, customizable data augmentations, real-time inference via Gradio, and support for model export (ONNX & PyTorch).

Deep Learning training and deployment pipeline, reduce repetitive work from research to deployment

This repository features an image sharpening pipeline using Knowledge Distillation. A high-capacity Restormer acts as the teacher model, while a lightweight Mini-UNet is trained as the student to mimic its performance.

End-to-end machine learning project for diamond price prediction using Lasso Regression with 93% accuracy.

Training an image classification model with CIFAR-10 dataset

🧠 Deep-Learning Evolution: Unified collection of TensorFlow & PyTorch projects, featuring custom CUDA kernels, distributed training, memory‑efficient methods, and production‑ready pipelines. Showcases advanced GPU optimizations, from foundational models to cutting‑edge architectures. 🚀

SPIRA Model Trainer v2 (redesigned pipeline) by @danlawand

PyTorch detailed analysis to create Machine learning to Deep learning model

Configurable PyTorch training pipeline

YoloLint is a tool for automatic validation of dataset structure, annotation files, and image sizes in YOLO projects. It helps you catch typical errors in directory structure, YAML files, annotation files, and now also ensures all your images have the correct size before you start model training.

AI Message Labels: Packaging and pipelines for deep learning text classification models

Machine Learning in Production

A comprehensive framework for experimenting with and comparing modern object detection models including YOLO (v5, v8) and Detectron2. Features automated setup, training pipelines, benchmarking tools, and Jupyter notebooks for computer vision research and development.

SPIRA Model Trainer v3 (productionized pipeline) by @lucasqml and @bolgheroni

Collection of Jupyter notebooks and python code files explaining the essentials of Pytorch.

end to end classification project , to check whether chicken is healthy or affeccted with coccidiosis based on chicken fecal image collected , Model used - VGG16

tracebloc notebook to launch and manage experiments in collaboration