mmcls/datasets/pipelines/transforms.py 中也没有这个类
时间: 2023-07-05 15:28:03 浏览: 95
YOLO 数据集:食材缺陷目标检测(10类)【包含划分好的数据集、类别class文件、数据可视化脚本】
非常抱歉,我的回答还是有误。在 mmclassification 0.25.0 版本中,`LoadImageFromFile` 类实际上是在 `mmcls/datasets/pipelines/compose.py` 中定义的。
默认情况下,`LoadImageFromFile` 类使用的是 PIL 库来读取图片。如果需要使用 OpenCV 读取图片,可以在配置文件中将 `to_rgb` 设置为 False,并将 `backend` 设置为 `opencv`.
例如,以下配置文件中使用的就是 OpenCV 来读取图片:
```python
data:
train:
pipeline:
- type: LoadImageFromFile
to_rgb: False
color_type: 'color'
backend: 'opencv'
- type: Resize
size: (256, 256)
- type: Normalize
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
- type: ToTensor
```
需要注意的是,使用 OpenCV 读取图片时,需要将图片的通道顺序设置为 BGR。
阅读全文