当直接下载GitHub上面的代码，并且在Pycharm中打开后，明明存在augmentation.py文件，但是无法引用其中的方法。

 解决方法：将所在的文件夹设置为Sources Root即可

 然后就不再报错了