A PyTorch-based Python package for deep and machine learning analysis of microscopy data, designed for domain scientists.
AtomAI is a PyTorch-based Python package designed for deep and machine learning analysis of microscopy data. It provides tools for semantic segmentation, spectral prediction, variational autoencoders, and deep kernel learning, enabling scientists to extract physical insights from complex image and hyperspectral datasets without advanced ML expertise.
Domain scientists and researchers in microscopy (e.g., materials science, biology) who have basic Python knowledge (NumPy, Matplotlib) and want to apply ML to their data without becoming ML experts.
It offers a simplified, intuitive interface that bridges powerful PyTorch models with scientific workflows, reducing the barrier to using state-of-the-art ML for microscopy analysis compared to building custom solutions from scratch.
Deep and Machine Learning for Microscopy
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
Offers simple interfaces like Segmentor.fit() and model.predict() that reduce code complexity, enabling training with just a few lines of code, as shown in the semantic segmentation example.
Includes specialized models for semantic segmentation, ImSpec prediction, and VAEs tailored for microscopy data analysis, bridging instrument libraries with physical insights.
Provides tools for deep ensembles and deep kernel learning, allowing for reliable predictions with uncertainty quantification, which is critical for scientific experimentation.
Allows custom PyTorch models to be used with AtomAI's trainers and predictors, offering flexibility while maintaining ease of use, as demonstrated in the custom denoising autoencoder example.
Requires PyTorch installation and assumes user comfort with it, creating a barrier for teams entrenched in other frameworks like TensorFlow or with limited deep learning infrastructure.
Optimized for microscopy data, so features may not translate well to other scientific or industrial imaging tasks without significant adaptation, as admitted in the focus on scientific data.
Despite simplified APIs, users still need to preprocess data into NumPy arrays and understand basic ML concepts, which can be challenging for absolute beginners in programming or data science.