[add] ConcatDataset.evaluate feature #1932
Conversation
|
|
|
@17862923747 Hi, thank you very much for your help! Could you please sign the CLA so we can accept your contribution? This PR will be reviewed asap. |
| elif len(set([type(ds) for ds in self.datasets])) != 1: | ||
| raise NotImplementedError( | ||
| 'All the datasets should have same types') |
There was a problem hiding this comment.
Do we need to perform this check if separate_eval=False is not supported?
| dataset if `self.separate_eval=True`. | ||
| """ | ||
|
|
||
| new_results, res_len = self.rebuild_results(results) |
There was a problem hiding this comment.
Could you please explain the reason for concatenating all elements into new_results rather than simply utilizing results?
There was a problem hiding this comment.
results是多个val数据集混在一起的，然后每batch_size个输出组成results中一个dict，我需要根据cumulative_sizes记录的索引分割不同val数据集对应的results，分别计算各自的evaluate，我觉得先合并成一个results，然后分割比较方便，不知道我的理解对不对
There was a problem hiding this comment.
谢谢，我疑惑的点是有没有可能直接在 clip_by_index 函数里面利用 start_idx 和 end_idx 从 results 中提取每个数据集对应的 predictions, 然后再 concat？这样子可以省去 rebuild_results 这个步骤
| clip_res['preds'] = results[0]['preds'][start_idx:end_idx, :, :] | ||
| clip_res['boxes'] = results[0]['boxes'][start_idx:end_idx, :] | ||
| clip_res['image_paths'] = results[0]['image_paths'][start_idx:end_idx] | ||
| clip_res['bbox_ids'] = results[0]['bbox_ids'][start_idx:end_idx] |
There was a problem hiding this comment.
Bottom-up models may not generate predictions with key bbox_ids, so a key check may be needed here
|
@17862923747 您好，请问您还有更新这个 PR 的计划吗？如果有任何困难，我们也可以帮助您一起完善这个 PR 的内容 |
|
Hi @17862923747 ！We are grateful for your efforts in helping improve this open-source project during your personal time. Welcome to join OpenMMLab Special Interest Group (SIG) private channel on Discord, where you can share your experiences, ideas, and build connections with like-minded peers. To join the SIG channel, simply message moderator— OpenMMLab on Discord or briefly share your open-source contributions in the #introductions channel and we will assist you. Look forward to seeing you there! Join us ：https://discord.gg/UjgXkPWNqA Thank you again for your contribution❤ |
Motivation
多数据集场景，val时会报错，因为您的ConcatDataset.evaluate没有实现
Modification
我补充了缺少的这部分代码：ConcatDataset.evaluate
BC-breaking (Optional)
可兼容已有代码和config
Use cases (Optional)
Checklist
Before PR:
After PR: