This repository contains the source code, training data and a pretrained model for "Automated Structure Discovery for Scanning Tunneling Microscopy"
https://pubs.acs.org/doi/10.1021/acsnano.3c12654
Pytorch is used to train the model. Formal requirement list will be added shortly.
Data and pretrained weights available at Zenodo
These can be downloaded (5 GB) with python load_data.py.