使用json标签在数据集中水平翻转图像的简单方法?

2024-08-31 08:49:35 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用Tensorflow 2.0和Python来训练图像分类器。我使用文件model_main_tf2.py来训练模型,并拥有一个用于训练和测试的图像数据集。图像是使用Python中的LabelMe工具进行注释的,该工具允许我为掩码RCNN创建多边形掩码

我想做的是通过水平翻转来生成所有训练和测试图像的副本。我已经可以在python中轻松地完成这项工作,但我希望翻转LabelMe生成的JSON文件,以避免重新注释新翻转的图像。有没有一个工具可以让我这样做

谢谢


Tags: 文件工具数据py模型图像model分类器
1条回答
网友
1楼 · 发布于 2024-08-31 08:49:35

因为这个问题在Python标签下——我假设您希望在Python中完成。翻转可以在numpyPILopencv(您的选择)中完成

image = # some image translated to a numpy array
print(type(image))
>> numpy.ndarray

# np.fliplr will also do the trick
flipped_image_h = np.flip(image, axis=1) # flip horizontally

# np.flipud will also do the trick
flipped_image_v = np.flip(image, axis=0) # flip vertically

# Save flipped_image

See the numpy docs for more info

相关问题 更多 >