YCB-LUMA
收藏YCB-LUMA 数据集
概述
YCB-LUMA 数据集用于2D物体检测的亮度键控测试数据集的处理脚本。
使用方法
脚本执行顺序
generate_input_dirs.py(可选)- 仅在需要创建新的文件夹或在没有输入视频的情况下运行。
generate_masks.py- 生成掩码和RGB图像,CPU使用率可能较高。
delete_files.py- 过滤并删除噪声较大或质量差的掩码。
count.py(可选)- 统计输入视频和生成的掩码数量。
脚本描述
generate_input_dirs.py
- 根据
data.csv生成所需的输入目录。 data.csv使用分号;作为分隔符。- 生成的文件夹路径为
<name of dataset>/in/<name of generated directory>。
generate_masks.py
- 从输入视频生成掩码和RGB图像。
- 输出目录自动生成。
- 生成包含视频ID和输入名称的单独CSV文件。
count.py
- 统计输入视频和生成的掩码数量。
- 结果保存为单独的CSV文件,使用分号
;作为分隔符。
delete_files.py
- 根据
files_to_delete.csv删除指定的输出文件。 - 文件名不需要排序。
引用
如果使用该数据集,请引用以下文献:
@misc{pöllabauer2024ycblumaycbobjectdataset, title={YCB-LUMA: YCB Object Dataset with Luminance Keying for Object Localization}, author={Thomas Pöllabauer}, year={2024}, eprint={2411.13149}, archivePrefix={arXiv}, primaryClass={cs.CV}, url={https://arxiv.org/abs/2411.13149}, } @inproceedings{pöllabauer2024fast, title={Fast Training Data Acquisition for Object Detection and Segmentation using Black Screen Luminance Keying}, author={ extbf{Thomas P{"o}llabauer} and Knauthe, Volker and Boller, André and Kuijper, Arjan and Fellner, Dieter}, booktitle={Journal of WSCG}, volume={32}, number={1-2}, pages={101-110}, year={2024}, } @inproceedings{poellabauer2024LUMA++, title={Advanced Post-Processing for Object Detection Dataset Generation}, author={ extbf{Thomas P{"o}llabauer} and Berkei, Sarah and Knauthe, Volker and Kuijper, Arjan}, series={Lecture Notes in Computer Science}, booktitle={Proceedings of the 19th International Symposium on Visual Computing (ISCV24)}, year={2024} }




