search

liamheng / Annotation-free-Fundus-Image-Enhancement

Project for annotation-free image enhancement of fundus, cataract data.
30 stars 5 forks source link

Hello, I would like to know why this error occurred. #5

Closed qihuiyan1023 closed 5 months ago

qihuiyan1023 commented 8 months ago

ModuleNotFoundError: No module named data.unaligned_dataset

liamheng commented 5 months ago

@HeverLaw

HeverLaw commented 5 months ago

This error is because you did not specify a dataset. In our code, you can use -- dataset_mode catalact_with_mask. Please enter the corresponding command before running the code, for example, if you want to train GFE Net, enter Python train.py -- dataroot/ images/cataract_dataset --name train_gfenet --model gfenet --direction AtoB --dataset_mode cataract_with_mask --norm instance --batch_size 8 --gpu_ids 2 --lr_policy linear --n_epochs 150 --n_epochs_decay 50.