search

Linfeng-Tang / SeAFusion

The code of " Image fusion in the loop of high-level vision tasks: A semantic-aware real-time infrared and visible image fusion network"
MIT License
184 stars 34 forks source link

about train_seg #49

Open Joanna006 opened 1 month ago

Joanna006 commented 1 month ago

您好，在融合图像完成后，遇到的问题 |1 Fusion Image Sucessfully~! ./MSRS/Fusion/train ./MSRS/Fusion/train/MSRS ./MSRS/Label/train/MSRS Traceback (most recent call last): File "train.py", line 404, in train_seg(i, logger, args) File "train.py", line 87, in train_seg ds = CityScapes('./MSRS/', cropsize=cropsize, mode='train', Method=Method) File "/root/autodl-tmp/SeAFusion-main/cityscapes.py", line 71, in init assert set(imgnames) == set(gtnames) AssertionError

Linfeng-Tang commented 1 month ago

可以尝试一下修改Dataloader即可。