OpenViSTAI · Agents · Blockchain · Crypto
Submit a project
Menu
AI Developer Tools & SDKs

Nitrain

Nitrain is an open-source Python library for sampling and augmenting medical images, preparing medical-imaging datasets, training models, and visualizing results. It provides high-level training utilities and supports PyTorch, TensorFlow, and Keras backends.

Explore the practical guide ↓

Independent OpenViST project profile, based on linked sources.

Nitrain — OpenViST project profile in AI Developer Tools & SDKs.
OpenViST project illustration. See the linked sources for product details.
Understand the project

Practical overview

When it fits

Use Nitrain when a Python medical-imaging project needs dataset, sampling, augmentation, and training utilities in one workflow, with a choice of PyTorch, TensorFlow, or Keras.

Who it is for

  • Python developers building medical-imaging AI workflows
  • Researchers preparing medical-image datasets for segmentation, classification, or registration models

Documented uses

  • Read medical images and organize model inputs and target labels into a dataset.
  • Apply image transforms, sampling, and augmentation before training.
  • Train a medical-imaging model using a supported framework backend.
  • Visualize medical-image data and results with ANTsPy support.
Before you choose

Requirements & limits

  • ANTsPy is identified as a key dependency for reading, operating on, and visualizing medical images.
  • Model creation uses a supported backend: PyTorch, TensorFlow, or Keras.
  • The repository is public and specifies the AGPL-3.0 license.

Check current requirements in the official documentation. Source-based guidance does not mean OpenViST has run or independently tested this project.

Source-based practical guide

Set up Nitrain for medical-image model training

Install the library and identify the image-handling dependency and model backend required by the documented workflow.

Before you start

  • Python and pip
  • ANTsPy for medical-image reading, operations, and visualization
  • A model backend: PyTorch, TensorFlow, or Keras

Steps

  1. Install the latest release with the command below.

    Documented command
    pip install nitrain

  2. Install or configure ANTsPy for medical-image handling.

  3. Choose a supported backend for creating models.

Expected outcome

Nitrain is installed, with its documented medical-image dependency and a backend selected for model creation.

Guide limitations

  • This setup guide does not cover backend-specific installation steps.
  • Installation instructions are taken from project documentation and are not execution tests.
Documentation 1 ↗
Evidence and transparency

Sources & review

Official repository ↗
Source record details
Source availability
Active
Sources checked
15 October 2026
Original website title
nitrain/nitrain · Official homepage title · retained in its source language
How we review sourcesSuggest a correction