no module named 'augmentation'

SeuTao / FaceBagNet

FaceBagNet - Patch-based Methods for Multi-modal Face Anti-spoofing (FAS)

664 stars 201 forks source link

no module named 'augmentation' #25

Closed Homura2333 closed 3 years ago

Homura2333 commented 3 years ago

train using CUDA_VISIBLE_DEVICES=0 python train_Fusion_CyclicLR.py --model=model_A --image_size=48

KiLJ4EdeN commented 3 years ago

train using CUDA_VISIBLE_DEVICES=0 python train_Fusion_CyclicLR.py --model=model_A --image_size=48

Do the imports like this from .augmentation import *, this way the modules are imported from the same directory. This is probably because you are using an IDE. You could do the same for similar import errors.