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
207 stars 37 forks source link

about train_seg #49

Open Joanna006 opened 4 months ago

Joanna006 commented 4 months 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 4 months ago

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

wyfARTHUR commented 1 month ago

您好，请问问题解决了吗？具体是如何修改的？