验证码自动识别-1

#include <dlib/dnn.h> #include <dlib/opencv.h> #include <dlib/image_io.h> #include <bitset> #include <dlib/misc_api.h> #include <explore.h> #include <opencv2/opencv.hpp> using namespace std; using namespace dlib; using namespace cv; using net_type = loss_multiclass_log< fc<10, relu<fc<128, avg_pool_everything< relu<affine<con<32,3,3,1,1, max_pool<3,3,2,2,relu<affine<con<16,3,3,1,1, input_rgb_image >>>>>>>>>>>>; void get_crop(const Mat& image,std::vector<matrix<rgb_pixel>>& crop_images) { Mat crop1 = Mat(image,Range(5,17),Range(9,18)).clone(); Mat crop2 = Mat(image,Range(5,17),Range(18,27)).clone(); Mat crop3 = Mat(image,Range(5,17),Range(27,36)).clone(); Mat crop4 = Mat(image,Range(5,17),Range(36,45)).clone(); resize(crop1,crop1,Size(28,28),0,0,INTER_NEAREST); resize(crop2,crop2,Size(28,28),0,0,INTER_NEAREST); resize(crop3,crop3,Size(28,28),0,0,INTER_NEAREST); resize(crop4,crop4,Size(28,28),0,0,INTER_NEAREST); matrix<rgb_pixel> img1,img2,img3,img4; assign_image(img1,cv_image<rgb_pixel>(crop1)); assign_image(img2,cv_image<rgb_pixel>(crop2)); assign_image(img3,cv_image<rgb_pixel>(crop3)); assign_image(img4,cv_image<rgb_pixel>(crop4)); crop_images.push_back(img1); crop_images.push_back(img2); crop_images.push_back(img3); crop_images.push_back(img4); } int main(int argc,char** argv) { if(argc != 2) { cout<<"invalid input! 2 arguments are expected."; return 1; } string dir = argv[1]; int index = 0; for(string file : directory(dir).get_files()) { Mat image = imread(file); Mat crop1 = Mat(image,Range(5,17),Range(9,18)); Mat crop2 = Mat(image,Range(5,17),Range(18,27)); Mat crop3 = Mat(image,Range(5,17),Range(27,36)); Mat crop4 = Mat(image,Range(5,17),Range(36,45)); resize(crop1,crop1,Size(28,28),0,0,INTER_NEAREST); resize(crop2,crop2,Size(28,28),0,0,INTER_NEAREST); resize(crop3,crop3,Size(28,28),0,0,INTER_NEAREST); resize(crop4,crop4,Size(28,28),0,0,INTER_NEAREST); matrix<rgb_pixel> img1,img2,img3,img4; assign_image(img1,cv_image<rgb_pixel>(crop1)); assign_image(img2,cv_image<rgb_pixel>(crop2)); assign_image(img3,cv_image<rgb_pixel>(crop3)); assign_image(img4,cv_image<rgb_pixel>(crop4)); cout<<file<<endl; string train = "train/"; string name1 = train + file[37] + "/" + to_string(index) + ".png"; save_png(img1,name1); index++; string name2 = train + file[38] + "/" + to_string(index) + ".png"; save_png(img2,name2); index++; string name3 = train + file[39] + "/" + to_string(index) + ".png"; save_png(img3,name3); index++; string name4 = train + file[40] + "/" + to_string(index) + ".png"; save_png(img4,name4); index++; } }