OpticsBench, LensCorruptions
收藏数据集概述:分类模型对常见光学畸变的鲁棒性研究
数据集基本信息
- 作者:Patrick Müller (2023)
- 相关论文:ICCV2023 AROW Workshop官方提交论文
- 许可证:参见/opticsbench和/opticsaugment目录
数据集内容
核心组件
-
OpticsBench
- 用于研究对现实光学模糊效应的鲁棒性基准
- 包含基于Zernike多项式的光学畸变类型:彗差、散光、球差、三叶草像差
- 支持生成预定义的图像损坏数据集
-
OpticsAugment
- 使用光学核的数据增强方法
- 可提高模型对光学畸变和2D常见损坏的鲁棒性
代码结构
/opticsbench:创建模糊损坏数据集和评估PyTorch DNN的代码/opticsaugment:使用OpticsAugment训练模型的代码
性能指标
OpticsBench ImageNet-100结果(平均准确率%)
| 模型 | 1 | 2 | 3 | 4 | 5 |
|---|---|---|---|---|---|
| DenseNet(ours) | 68.22 | 65.33 | 56.33 | 41.60 | 30.13 |
| EfficientNet | 61.00 | 55.34 | 42.14 | 30.27 | 23.35 |
| MobileNet | 57.59 | 52.30 | 38.58 | 27.51 | 20.54 |
| ResNet101 | 69.90 | 67.68 | 61.36 | 49.04 | 37.80 |
| ResNeXt50 | 65.14 | 62.68 | 54.44 | 39.90 | 28.45 |
常见损坏性能提升(百分点)
| 模型 | 1 | 2 | 3 | 4 | 5 |
|---|---|---|---|---|---|
| DenseNet161 | 5.08 | 7.55 | 8.73 | 7.30 | 5.38 |
| ResNeXt50 | 5.11 | 7.63 | 8.68 | 7.18 | 5.27 |
| ResNet101 | 1.25 | 3.07 | 4.55 | 4.90 | 4.10 |
使用方法
数据集生成
bash python benchmark.py --generate_datasets --database imagenet-1k_val --testdata_path <path_to_validation_images>
生成路径结构:data/images/<dataset>/<val,corruptions>/<corruption_name>/<severity>/
模型评估
bash python benchmark.py --run_all --path_to_root_folder <root> --models all
结果输出路径:root/eval/<dataset>/corruptions>/<corruption_name>/<severity>/<model_name>.json
引用格式
bibtex @InProceedings{Muller_2023_ICCV, author = {M"uller, Patrick and Braun, Alexander and Keuper, Margret}, title = {Classification Robustness to Common Optical Aberrations}, booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV) Workshops}, month = {October}, year = {2023}, pages = {3632-3643} }
文件目录结构示例
root/ images/ ImageNette/ /val /corruptions/ opticsblur/ astigmatism/ 1/ 2/ ... eval/ ImageNette/ /val /corruptions/ opticsblur/ astigmatism/ 1/ resnet50.json ... models/ resnet50_augmix_optics_augment/ ...




