assert len(label_class) > 0

[Carina-HH]

您好！在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]

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

[Carina-HH]

重新搞数据集

…

---- 回复的原邮件 ---- | 发件人 | ***@***.***> | | 日期 | 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: ***@***.***>

[qjmltz]

重新搞数据集
…
---- 回复的原邮件 ---- | 发件人 | @.> | | 日期 | 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]

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

[Carina-HH]

应该是数据集中某几张图片的标注有问题

…

---- 回复的原邮件 ---- | 发件人 | ***@***.***> | | 日期 | 2024年12月02日 10:54 | | 收件人 | ***@***.***> | | 抄送至 | ***@***.***>***@***.***> | | 主题 | Re: [chunbolang/BAM] assert len(label_class) > 0 (Issue #62) | 你好，我是在coco数据集上碰到了这个问题，请问重新搞数据集具体是指？ — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>

[Carina-HH]

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

…

---- 回复的原邮件 ---- | 发件人 | ***@***.***> | | 日期 | 2024年12月02日 10:54 | | 收件人 | ***@***.***> | | 抄送至 | ***@***.***>***@***.***> | | 主题 | Re: [chunbolang/BAM] assert len(label_class) > 0 (Issue #62) | 你好，我是在coco数据集上碰到了这个问题，请问重新搞数据集具体是指？ — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: ***@***.***>

[LevisWei]

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

好的我再试试

[Bruce-W12138]

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

[liuqi018]

您好！在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
我也是这个问题，请问最后怎么解决的？

[zhuhui1214]

有可能是生成coco数据的时候，coco数据的categories是1-90的，生成了类别数就是1-90，但是coco只有80类，90个类别中有些类别是有问题的，因此需要一个类别映射表，将1-90映射到1-80.coco_categories = [
1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 13, 14, 15, 16, 17, 18, 19, 20, 21,
22, 23, 24, 25, 27, 28, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42,
43, 44, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61,
62, 63, 64, 65, 67, 70, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 84,
85, 86, 87, 88, 89, 90
]