Skip to content

Simple trame demonstrator using the MNIST dataset with XAITK for saliency analysis

License

Notifications You must be signed in to change notification settings

Kitware/trame-mnist

Repository files navigation

Trame MNIST

Example application using trame for exploring MNIST dataset in the context of AI training and XAI thanks to XAITK.

Installing

For the Python layer it is recommended to use conda to properly install the various ML packages.

conda setup on macOS

Go to conda documentation for your OS

brew install miniforge
conda init zsh

venv setup for AI

# Needed in order to get py3.9 with lzma
# PYTHON_CONFIGURE_OPTS="--enable-framework" pyenv install 3.9.9

conda create --name trame-mnist python=3.9
conda activate trame-mnist

# For development when inside repo
pip install -e .

# For testing (no need to clone repo)
pip install trame-mnist

Running the application

conda activate trame-mnist
trame-mnist

If cuda is available, the application will use your GPU, but you can also force the usage of your cpu by adding to your command line the following argument: --cpu

image_1 image_2 image_3

License

trame-mnist is distributed under the OSI-approved BSD 3-clause License.

About

Simple trame demonstrator using the MNIST dataset with XAITK for saliency analysis

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published