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: over 2 years ago
deep-learningkerasmedical-imagingneuroimagingpytorch

Related projects:

RepositoryDescriptionStars
wyy-123-xyy/ra-fedA Python implementation of a distributed machine learning framework for training neural networks on multiple GPUs6
nipy/nibabelA package that provides access to various neuro-imaging file formats through a unified interface.660
intelligent-machine-learning/dlroverAutomates large-scale deep learning training on distributed clusters, providing fault tolerance and fast recovery from failures.1,302
neuralhydrology/neuralhydrologyA Python library for training neural networks with focus on hydrological applications using PyTorch.372
martinkersner/train-deeplabTrains DeepLab model for semantic image segmentation using annotated data and various training procedures172
huggingface/nanotronA pretraining framework for large language models using 3D parallelism and scalable training techniques1,332
ahmedfgad/cnngeneticTrains CNNs using a genetic algorithm for classification problems where one class per sample is allowed.22
adalca/neuriteA toolbox for building neural networks that analyze medical images340
open-mmlab/mmengineProvides a flexible and configurable framework for training deep learning models with PyTorch.1,196
truskovskiyk/nima.pytorchAssesses and evaluates images using deep learning models335
ahmedfgad/neuralgeneticTrains artificial neural networks using the genetic algorithm241
ethanhe42/u-netA convolutional neural network architecture for biomedical image segmentation430
nicrusso7/rex-gymTraining an open-source quadruped robot to perform domestic and generic tasks using reinforcement learning and OpenAI Gym.998
achaiah/pywickA PyTorch-based neural network training framework with advanced features and utilities398
ahmedfgad/kerasgaTrains Keras models using the Genetic Algorithm36