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: 5 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. 654
intelligent-machine-learning/dlrover An automatic distributed deep learning system that simplifies the training of large AI models 1,270
neuralhydrology/neuralhydrology A Python library for training neural networks with focus on hydrological applications using PyTorch. 364
martinkersner/train-deeplab Trains DeepLab for semantic image segmentation using annotated data and convolutional neural networks 172
huggingface/nanotron A library for training large language models with parallel computing and mixed precision training methods 1,244
ahmedfgad/cnngenetic Trains CNNs using the genetic algorithm 22
adalca/neurite A collection of tools and models for building neural networks for medical image analysis 340
open-mmlab/mmengine Provides a flexible and configurable framework for training deep learning models with PyTorch. 1,179
truskovskiyk/nima.pytorch Assesses and evaluates images using deep learning models 335
ahmedfgad/neuralgenetic Trains artificial neural networks using the genetic algorithm to optimize regression or classification tasks 241
ethanhe42/u-net A convolutional neural network architecture for biomedical image segmentation 426
nicrusso7/rex-gym Training an open-source quadruped robot to perform domestic and generic tasks using reinforcement learning and OpenAI Gym. 988
achaiah/pywick A PyTorch-based neural network training framework with advanced features and utilities 398
ahmedfgad/kerasga Trains Keras models using a genetic algorithm optimization method 36