在机器学习训练的过程中,可能会因为数据量少,或者数据过于集中,造成模型效果不是很好,或者容易过拟合.

我们可以使用cv2.flip()将图片进行反转,这样能够增加数据量,也可以让模型更加对称.

cv2.flip(src, flipCode[, dst])
  • src:要反转的数据
  • flipCode:为1时水平翻转;为0时垂直翻转;为-1时中心对称.
# coding=utf-8
import matplotlib.pyplot as plt
import numpy as np
import cv2

img = plt.imread('./cat.jpg')
newimg = cv2.flip(img,1)

f,(ax1,ax2) = plt.subplots(1,2)
ax1.imshow(img)
ax2.imshow(newimg)
plt.show()

posted @ 2018-05-16 16:00:11
评论加载中...

发表评论