atomai
atomai copied to clipboard
pycroscopy
Deep and Machine Learning for Microscopy
**Function:** create_lattice_mask **File:** imgen.py **Issue:** XY coordinates close an edge can cause an error if their gaussian extends past the boundary. **Potential Solution:** This can be avoided by including cases...
Line 357 in trainer.py **gpu_usage = gpu_usage_map(torch.cuda.current_device())** may result in FileNotFoundError. Work-around is to just use Try Except block to bypass error. Windows may view as unsafe command.
The [semantic segmentation models](https://github.com/pycroscopy/atomai/blob/master/atomai/nets/fcnn.py) need to be extended to work with 3D data. This should be very straightforward - just introduce an option to select between 1D, 2D, and 3D...
### pycroscopy/atomai now has a Chat Room on Gitter @ziatdinovmax has just created a chat room. You can visit it here: [https://gitter.im/pycroscopy/AtomAI](https://gitter.im/pycroscopy/AtomAI?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&content=body_link). This pull-request adds this badge to your README.md:...
We may want to add the following neural network architecture to nets/fcnn.py module: https://www.pnas.org/content/115/2/254 @aghosh92 Can you please take a look at it?
Currently, we have rather archaic [data augmentation pipelines](https://github.com/pycroscopy/atomai/blob/master/atomai/transforms/imaug.py), which employ a combination of scikit-learn and open-cv image processing functions. This leads to a significant slow down of the training with...
Are you planning to open a discussion section? I have something to discuss. Shall I continue here?
Passing a custom data loader in AtomAI models can be a useful feature. This would allow using e.g. Kornia data augmentation pipelines. It could look like this ```python3 segmodel =...
I installed atomai in a conda python 3.11 environment, and I got the following when attempting to import the package: ``` Python 3.11.5 (main, Sep 11 2023, 13:54:46) [GCC 11.2.0]...
3D fcnn
Updated fcnn blocks with 3D versions. I am wondering if separate 3D versions of fcnns should be created or if existing fcnns should be updated with ndim arg in fcnn.py.
