Imported from https://github.com/ssea-lab/DL4ETI