nitrain

Medical imaging AI trainer

A framework-agnostic Python library for training AI models on medical images

Train AI models efficiently on medical images using any framework

GitHub

2k stars
50 watching
302 forks
Language: Python
last commit: 7 months ago
deep-learningkerasmedical-imagingneuroimagingpytorch

Related projects:

Repository Description Stars
wyy-123-xyy/ra-fed A Python implementation of a distributed machine learning framework for training neural networks on multiple GPUs 6
nipy/nibabel A package that provides access to various neuro-imaging file formats through a unified interface. 660
intelligent-machine-learning/dlrover Automates large-scale deep learning training on distributed clusters, providing fault tolerance and fast recovery from failures. 1,302
neuralhydrology/neuralhydrology A Python library for training neural networks with focus on hydrological applications using PyTorch. 372
martinkersner/train-deeplab Trains DeepLab model for semantic image segmentation using annotated data and various training procedures 172
huggingface/nanotron A pretraining framework for large language models using 3D parallelism and scalable training techniques 1,332
ahmedfgad/cnngenetic Trains CNNs using a genetic algorithm for classification problems where one class per sample is allowed. 22
adalca/neurite A toolbox for building neural networks that analyze medical images 340
open-mmlab/mmengine Provides a flexible and configurable framework for training deep learning models with PyTorch. 1,196
truskovskiyk/nima.pytorch Assesses and evaluates images using deep learning models 335
ahmedfgad/neuralgenetic Trains artificial neural networks using the genetic algorithm 241
ethanhe42/u-net A convolutional neural network architecture for biomedical image segmentation 430
nicrusso7/rex-gym Training an open-source quadruped robot to perform domestic and generic tasks using reinforcement learning and OpenAI Gym. 998
achaiah/pywick A PyTorch-based neural network training framework with advanced features and utilities 398
ahmedfgad/kerasga Trains Keras models using the Genetic Algorithm 36