从数据集随机选取80个图像,定义字典为图像路径和标签,最终的数据集是由字典组成的列表,且每次随机选取的图像不一样
时间: 2024-03-12 19:45:21 浏览: 50
好的,您可以使用以下 Python 代码来实现这个功能:
```python
import os
import random
# 定义图像文件夹路径
image_folder = "/path/to/image/folder"
# 定义标签列表
labels = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
# 获取图像文件夹中所有图像文件的路径
image_paths = [os.path.join(image_folder, f) for f in os.listdir(image_folder) if os.path.isfile(os.path.join(image_folder, f))]
# 随机选择图像文件,并将其路径和标签组成字典
data = []
for i in range(80): # 选取 80 张图像
random_index = random.randint(0, len(image_paths)-1)
random_image = image_paths[random_index]
random_label = random.choice(labels)
data.append({"image_path": random_image, "label": random_label})
# 从图像列表中删除已经选取的图像,确保每次选择的图像不重复
image_paths.pop(random_index)
# 输出数据
print(data)
```
这段代码会从 `image_folder` 文件夹中随机选取 80 张图像,并将图像文件路径和标签组成字典,存储在一个列表中。每次选取的图像不一样,确保数据集的多样性。您可以根据自己的需求修改代码中的参数。
阅读全文