论文标题

Atomvision:原子图像的机器视觉库

AtomVision: A machine vision library for atomistic images

论文作者

Choudhary, Kamal, Gurunathan, Ramya, DeCost, Brian, Biacchi, Adam

论文摘要

计算机视觉技术在材料设计应用中具有巨大的潜力。在这项工作中,我们引入了一个集成的通用原子库库,该库可用于生成,策展扫描隧道显微镜(STM)和扫描传输电子显微镜(STEM)数据集并应用机器学习技术。 To demonstrate the applicability of this library, we 1) generate and curate an atomistic image dataset of about 10000 materials, 2) develop and compare convolutional and graph neural network models to classify the Bravais lattices, 3) develop fully convolutional neural network using U-Net architecture to pixelwise classify atom vs background, 4) use generative adversarial network for super-resolution, 5) curate a natural language processing based image dataset使用开放式ARXIV数据集,6)将计算框架与实验显微镜工具集成在一起。 Atomvision库可从https://github.com/usnistgov/atomvision获得。

Computer vision techniques have immense potential for materials design applications. In this work, we introduce an integrated and general-purpose AtomVision library that can be used to generate, curate scanning tunneling microscopy (STM) and scanning transmission electron microscopy (STEM) datasets and apply machine learning techniques. To demonstrate the applicability of this library, we 1) generate and curate an atomistic image dataset of about 10000 materials, 2) develop and compare convolutional and graph neural network models to classify the Bravais lattices, 3) develop fully convolutional neural network using U-Net architecture to pixelwise classify atom vs background, 4) use generative adversarial network for super-resolution, 5) curate a natural language processing based image dataset using open-access arXiv dataset, and 6) integrate the computational framework with experimental microscopy tools. AtomVision library is available at https://github.com/usnistgov/atomvision.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源