数据增强方法:图片镜像、图片缩放、图片旋转、加噪点

1、图片镜像

//rotate with mirror
#include <iostream>  
#include <opencv2/opencv.hpp>  
using namespace std;
using namespace cv;
int main()
{
    int d = 0;
    int dx = 0;
    char image[100];
    char ad[1000];
    for (int i = 0; i < 18; i++) 
    {
        sprintf_s(image, "C:\\Users\\Desktop\\vechicle\\%d.jpg", ++d);
        Mat src = imread(image, CV_LOAD_IMAGE_UNCHANGED);
        Mat dst, map_x, map_y;
        dst.create(src.size(), src.type());
        map_x.create(src.size(), CV_32FC1);
        map_y.create(src.size(), CV_32FC1);

        for (int i = 0; i < src.rows; ++i)
        {
            for (int j = 0; j < src.cols; ++j)
            {
                //map_x.at<float>(i, j) = (float)(src.cols - j);
                //map_y.at<float>(i, j) = (float)i;  //水平
                map_x.at<float>(i, j) = (float)j;
                map_y.at<float>(i, j) = (float)(src.rows - i);  //垂直                                 
            }
        }

        remap(src, dst, map_x, map_y, CV_INTER_LINEAR);   //重映射,OpenCV函数

        sprintf_s(ad, "C:\\Users\\Desktop\\novehicle\\%d.jpg", ++dx);
        imwrite(ad, dst);
    }
    return 0;
}

remap函数说明:OpenCV:remap()简单重映射_SSS_369-CSDN博客_cv::remap

2、图片缩放

//resize the image
#include<iostream>
#include<opencv2\opencv.hpp>
using namespace cv;
using namespace std;
int main()
{
    Mat srcImage, dstImage, temp;
    int d = 0, dx = 0;
    char ad[3000];
    char image[100];
    for (int i = 0; i < 15; i++)
    {
        sprintf_s(image, "C:\\Users\\Desktop\\novehicle\\%d.jpg", d++);
        srcImage = imread(image);
        temp = srcImage;

        //图像缩放
        //resize(temp, dstImage, Size(temp.cols/8 , temp.rows/8 ), (0, 0), (0, 0), 3);
        resize(temp, dstImage, Size(128, 128), (0, 0), (0, 0), 3);

        sprintf_s(ad, "C:\\Users\\Desktop\\128wu\\%d.jpg", ++dx);
        imwrite(ad,dstImage);
    }

    return 0;
}

3、图片旋转

//rotate the image
#include<opencv2\opencv.hpp>
#include<iostream>
#include<stdio.h>
using namespace std;
using namespace cv;
int main() {
    cv::Mat dst;
    int dx = 0, d = 1;
    char ad[3000];
    char image[100];
    for (int i = 0; i < 81; i++)
    {
        sprintf_s(image, "C:\\Users\\Desktop\\novechicle\\%d.jpg", d++);
        Mat src = imread(image);
        for (double angle = 0; angle <= 360; angle = angle + 8) 
        {
            cv::Size dst_sz(src.cols, src.rows);
            cv::Point2f center(src.cols / 2., src.rows / 2.);
            cv::Mat rot_mat = cv::getRotationMatrix2D(center, angle, 1.0);
            cv::warpAffine(src, dst, rot_mat, dst_sz, INTER_LINEAR, BORDER_REPLICATE);

            sprintf_s(ad, "C:\\Users\\Desktop\\resize\\%d.jpg", ++dx);
            imwrite(ad, dst);
        }
    }

    return 0;
}

4、添加噪声

//add noise
#include <opencv2/opencv.hpp>  
#include <opencv2/highgui/highgui.hpp>  
#include<opencv2/imgproc/imgproc.hpp>
using namespace cv;
using namespace std;
void salt(Mat img, int n);
int main()
{
    int d = 0;
    int dx = 0;
    char image[1000];
    char ad[1000];
    for (int j = 0; j < 18; j++)
    {
        sprintf_s(image, "C:\\Users\\Desktop\\yuantu\\%d.jpg",++d);
        Mat img = imread(image);

        salt(img, 800);   //添加椒盐噪声

        sprintf_s(ad, "C:\\Users\\Desktop\\novehicle\\%d.jpg",++dx);
        imwrite(ad,img);
    }

    return 0;
}

void salt(Mat img, int n)
{
    for (int k = 0; k<n; k++)
    {
        int i = rand() % img.cols;
        int j = rand() % img.rows;

        if (img.channels() == 1)
        {
            img.at<uchar>(j, i) = 255;
        }
        else if (img.channels() == 3)
        {
            img.at<Vec3b>(j, i)[0] = 255;
            img.at<Vec3b>(j, i)[1] = 255;
            img.at<Vec3b>(j, i)[2] = 255;
        }
    }
}

原文:常用的数据增强方法——图片镜像、图片缩放、图片旋转、加噪点_syyyy712的博客-CSDN博客_数据增强 镜像

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值