Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

assert len(label_class) > 0 #62

Open
Carina-HH opened this issue Mar 2, 2024 · 8 comments
Open

assert len(label_class) > 0 #62

Carina-HH opened this issue Mar 2, 2024 · 8 comments

Comments

@Carina-HH
Copy link

您好!在pascal数据集上训练train.py时,在验证阶段报错 assert len(label_class) > 0,具体如下:
Traceback (most recent call last):
File "train.py", line 574, in
main()
File "train.py", line 234, in main
loss_val, FBIoU, FBIoU_m, mIoU, mIoU_m, mIoU_b, pIoU = validate(val_loader, model)
File "train.py", line 443, in validate
for i, (input, target, target_b, s_input, s_mask, subcls, ori_label, ori_label_b) in enumerate(val_loader):
File "/root/miniconda3/envs/py38/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 530, in next
data = self._next_data()
File "/root/miniconda3/envs/py38/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1204, in _next_data
return self._process_data(data)
File "/root/miniconda3/envs/py38/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1250, in _process_data
data.reraise()
File "/root/miniconda3/envs/py38/lib/python3.8/site-packages/torch/_utils.py", line 457, in reraise
raise exception
AssertionError: Caught AssertionError in DataLoader worker process 2.
Original Traceback (most recent call last):
File "/root/miniconda3/envs/py38/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "/root/miniconda3/envs/py38/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/miniconda3/envs/py38/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/autodl-fs/BAM_project/BAM-main/util/dataset.py", line 252, in getitem
assert len(label_class) > 0
AssertionError
个人判断,是因为验证时,pascal中某张图片的类别只属于sub_list,没有属于sub_val_list中的类别,因此label_class为空,因此提示异常。请问这是什么原因?应该怎么修改?非常感谢!

@qjmltz
Copy link

qjmltz commented May 22, 2024

兄弟你解决了吗?
这个问题我也遇到了不过中间的是
AssertionError: Caught AssertionError in DataLoader worker process 0.
最后也是
assert len(label_class) > 0
AssertionError
请问你有解决办法了吗?

@Carina-HH
Copy link
Author

Carina-HH commented May 22, 2024 via email

@qjmltz
Copy link

qjmltz commented May 22, 2024

重新搞数据集

---- 回复的原邮件 ---- | 发件人 | @.> | | 日期 | 2024年05月22日 18:23 | | 收件人 | @.> | | 抄送至 | @.>@.> | | 主题 | Re: [chunbolang/BAM] assert len(label_class) > 0 (Issue #62) | 兄弟你解决了吗? 这个问题我也遇到了不过中间的是 AssertionError: Caught AssertionError in DataLoader worker process 0. 最后也是 assert len(label_class) > 0 AssertionError 请问你有解决办法了吗? — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>

非常感谢,已经跑通了,感谢大佬指点

@LevisWei
Copy link

LevisWei commented Dec 2, 2024

你好,我是在coco数据集上碰到了这个问题,请问重新搞数据集具体是指?

@Carina-HH
Copy link
Author

Carina-HH commented Dec 2, 2024 via email

@Carina-HH
Copy link
Author

Carina-HH commented Dec 2, 2024 via email

@LevisWei
Copy link

LevisWei commented Dec 2, 2024

应该是数据集中图片标记有问题

---- 回复的原邮件 ---- |发件人 |@.> | |日期 |2024年12月02日 10:54 | |收件人 |@.> | |抄送至 |@.>@.> | |主题 |回复:[chunbolang/BAM] assert len(label_class) > 0 (Issue #62) | 你好,我是在coco数据集上碰到了这个问题,请问重新搞数据集具体是指? — 直接回复此电子邮件,在 GitHub 上查看,或取消订阅。 您收到此消息是因为您编写了该线程。消息 ID:@.***>

好的我再试试

@Bruce-W12138
Copy link

您好作者,我也遇到了这个问题,我用的是Pascal。不管是用util里面的脚本生成的base_annotation. 还是您给出的直接下载base_annotation的link似乎都没办法解决。总会有不在那20个类的数据,想问问能不能解决呢

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants