demo.cpp 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524
  1. // demo.cpp : 定义控制台应用程序的入口点。
  2. //
  3. #include "stdafx.h"
  4. #include <iostream>
  5. #include <opencv2\highgui\highgui.hpp>
  6. #include <opencv2\imgproc\imgproc.hpp>
  7. #include <opencv2\opencv.hpp>
  8. #include <map>
  9. #include <fstream>
  10. #include <time.h>
  11. #include <iomanip>
  12. #include <fstream>
  13. #include <stdlib.h>
  14. #include "utils.h"
  15. #include "config.h"
  16. #include "data_def.h"
  17. #include "optimal_angle.h"
  18. #include "cut_point_sc.h"
  19. #include "cut_point_rs.h"
  20. #include "graft_cv_api.h"
  21. #include "logger.h"
  22. using namespace cv;
  23. using namespace graft_cv;
  24. ofstream logger_ofs;
  25. CGcvLogger g_logger = CGcvLogger(
  26. logger_ofs,
  27. CGcvLogger::file_and_terminal,
  28. CGcvLogger::info,
  29. "D:\\logs\\gcv.log");
  30. void test_init_cp(ConfigParam&cp){
  31. cp.image_show=true;//
  32. cp.image_return=true;//
  33. cp.image_row_grid=20;
  34. cp.image_col_grid=100;
  35. cp.timeout_append=100;
  36. cp.timeout_proc=500;
  37. cp.self_camera=true;
  38. cp.image_save=true;
  39. cp.image_depository="D:\\logs\\img_depo";
  40. cp.image_backup_days=7;
  41. cp.oa_y_flip=true;
  42. cp.oa_morph_radius = 1;
  43. cp.oa_morph_iteration=2;
  44. cp.oa_min_hist_value=10;
  45. cp.oa_morph_radius_base = 1;
  46. cp.oa_morph_iteration_base=2;
  47. cp.oa_min_hist_value_base=10;
  48. cp.oa_col_th_ratio=0.7;
  49. cp.oa_row_th_ratio=1.2;
  50. cp.oa_stem_x_padding=100;
  51. cp.oa_stem_dia_min=10;
  52. cp.oa_stem_fork_y_min=10;
  53. cp.oa_stem_dia_mp=0.9;
  54. cp.oa_clip_y_min=245;
  55. cp.oa_clip_y_max = 355;
  56. cp.rs_y_flip=false;
  57. cp.rs_col_th_ratio= 0.7;
  58. cp.rs_row_th_ratio= 1.15;
  59. cp.rs_stem_x_padding= 40;
  60. cp.rs_stem_dia_min=12;
  61. cp.rs_stem_fork_y_min=10;
  62. cp.rs_stem_dia_mp=0.98;
  63. cp.rs_stem_edge_detect_window=5;
  64. cp.rs_morph_radius=1;
  65. cp.rs_morph_iteration=2;
  66. cp.rs_morph_iteration_gray=5;
  67. cp.rs_max_corner_num=500;
  68. cp.rs_corner_qaulity_level= 0.1;
  69. cp.rs_corner_min_distance= 10;
  70. cp.rs_cand_corner_box_width_ratio=3.0;
  71. cp.rs_cand_corner_box_xoffset_ratio=0.75;
  72. cp.rs_opt_corner_xoffset_ratio = 0.2;
  73. cp.rs_opt_corner_yoffset_ratio = -0.5;
  74. cp.rs_corner_mask_rad_ratio=0.25;
  75. cp.rs_cut_angle=-45;
  76. cp.sc_y_flip=false;
  77. cp.sc_col_th_ratio=0.7;
  78. cp.sc_row_th_ratio=2.5; //2-3.0
  79. cp.sc_stem_x_padding=50;
  80. cp.sc_stem_dia_min=3;
  81. cp.sc_clip_padding=5;
  82. cp.sc_stem_ymax_padding=50;
  83. cp.sc_default_cut_length=20;
  84. cp.sc_stem_edge_detect_window=5;
  85. cp.sc_r2_th=1.05;
  86. cp.sc_r2_window=10;
  87. cp.sc_average_window=10;
  88. cp.sc_morph_radius=1;
  89. cp.sc_morph_iteration=2;
  90. cp.rs_oa_pixel_ratio=1.0;
  91. cp.rs_cut_pixel_ratio=1.0;
  92. cp.sc_cut_pixel_ratio=1.0;
  93. }
  94. void test_imginfo2mat()
  95. {
  96. int t=0;
  97. ImgInfo* ii = new ImgInfo();
  98. ii->angle=0;
  99. ii->width = ii->height = 13;
  100. ii->data = new graft_cv::byte[ii->width*ii->height];
  101. for (int i=0;i<ii->height;++i){
  102. for (int j=0;j<ii->width;++j){
  103. if (i==j || i+j==ii->width-1){
  104. ii->data[i*ii->width+j]=255;
  105. }
  106. else{
  107. ii->data[i*ii->width+j]=0;
  108. }
  109. }
  110. }
  111. Mat img = imginfo2mat(ii);
  112. /*namedWindow("3_3", 0);
  113. imshow("3_3", img);
  114. waitKey(1);
  115. destroyAllWindows();*/
  116. delete [] ii->data;
  117. delete ii;
  118. };
  119. void test_camconfig_write()
  120. {
  121. ConfigParam cp0, cp1;
  122. memset(&cp1, 0, sizeof(ConfigParam));
  123. cp0 = cp1;
  124. CGCvConfig cam0 = CGCvConfig();
  125. CGCvConfig cam1 = CGCvConfig();
  126. cam0.setConfParam(&cp0);
  127. cam1.setConfParam(&cp1);
  128. FileStorage fs("cam_config.yml", FileStorage::WRITE);
  129. fs<<"conf_parameters";
  130. fs<<cam0;
  131. //fs<<cam1;
  132. //fs<<"]";
  133. fs.release();
  134. };
  135. void test_camconfig_read()
  136. {
  137. ConfigParam cp0, cp1;
  138. memset(&cp1, 0, sizeof(ConfigParam));
  139. cp0 = cp1;
  140. cp0.oa_min_hist_value = 100;
  141. cout<<&cp0<<endl;
  142. CGCvConfig cam0 = CGCvConfig();
  143. //CGCvConfig cam1 = CGCvConfig();
  144. cam0.setConfParam(&cp0);
  145. //cam1.setConfParam(cp1);
  146. FileStorage fs("cam_config.yml", FileStorage::READ);
  147. cam0.read(fs["conf_parameters"]);
  148. //fs<<cam1;
  149. //fs<<"]";
  150. fs.release();
  151. };
  152. void test_anglefit_readdata(vector<map<int,int>>& data){
  153. string ifile = "E:\\projects\\grafting_robots\\py_code\\test.txt";
  154. ifstream ifs(ifile.c_str(), ifstream::in );
  155. data.clear();
  156. if( ifs.is_open()){
  157. string line;
  158. map<int,int> tmp;
  159. while(getline(ifs,line)){
  160. std::cout<<line<<std::endl;
  161. size_t found = line.find(",");
  162. if(found !=string::npos){
  163. //found
  164. string sub0 = line.substr(0,found);
  165. string sub1 = line.substr(found+1);
  166. int an = int(stod(sub0));
  167. int wi = stoi(sub1);
  168. if(an==0){
  169. tmp.clear();
  170. }
  171. tmp.insert(make_pair<int,int>(an,wi));
  172. if(an==180){
  173. data.push_back(tmp);
  174. }
  175. }
  176. }
  177. ifs.close();
  178. }
  179. };
  180. void test_anglefit()
  181. {
  182. ConfigParam cp;
  183. COptimalAnglePart opa = COptimalAnglePart(cp);
  184. vector<map<int,int>> data;
  185. //test_anglefit_readdata(data);
  186. map<int,int>tmp;
  187. /*tmp.insert(make_pair<int,int>(0,216));
  188. tmp.insert(make_pair<int,int>(30,189));
  189. tmp.insert(make_pair<int,int>(60,112));
  190. tmp.insert(make_pair<int,int>(90,151));
  191. tmp.insert(make_pair<int,int>(120,188));*/
  192. tmp.insert(make_pair<int,int>(0,315));
  193. tmp.insert(make_pair<int,int>(30,270));
  194. tmp.insert(make_pair<int,int>(60,218));
  195. tmp.insert(make_pair<int,int>(90,141));
  196. tmp.insert(make_pair<int,int>(120,127));
  197. data.push_back(tmp);
  198. for(vector<map<int,int>>::iterator it = data.begin(); it!=data.end(); ++it){
  199. map<int,int>& an2width = *it;
  200. double oa = opa.angle_fit(an2width);
  201. cout<<oa<<endl;
  202. }
  203. };
  204. void test_optimal_angle(){
  205. //string folder = "E:\\projects\\grafting_robots\\py_code\\tmp";
  206. string folder = "D:\\grafting_robot\\samples\\tmp";
  207. namedWindow("pic", 0);
  208. vector<cv::String>filenames;
  209. ConfigParam cp;
  210. test_init_cp(cp);
  211. /*cp.oa_min_hist_value=10;
  212. cp.oa_morph_iteration=2;
  213. cp.oa_morph_radius = 1;*/
  214. COptimalAngle opa(cp);
  215. for(int i = 1;i<=5;++i){
  216. for(int j = 0;j<=1;++j){
  217. ostringstream ostr;
  218. ostr<<i<<j;
  219. string subfold = ostr.str();
  220. string src_folder = folder+"\\p"+subfold;
  221. cv::glob(src_folder, filenames);
  222. opa.reset();
  223. PositionInfo posinfo;
  224. clock_t t,t0;
  225. t = clock();
  226. for(size_t idx=0; idx<filenames.size();++idx){
  227. //cout<<filenames[idx]<<endl;
  228. size_t found0 = filenames[idx].rfind("\\");
  229. size_t found1 = filenames[idx].rfind(".");
  230. int an = stoi(filenames[idx].substr(found0+1, found1-found0))*20;
  231. if(an >=200){
  232. an-=200;
  233. }
  234. Mat img = imread(filenames[idx], cv::IMREAD_GRAYSCALE);
  235. ImgInfo* imginfo = mat2imginfo(img);
  236. imginfo->angle = an;
  237. int obj_width=0;
  238. try{
  239. t0 = clock();
  240. obj_width = opa.append(imginfo,posinfo);
  241. t0 = clock() - t0;
  242. imginfo_release(&imginfo);
  243. }
  244. catch(int i)
  245. {
  246. cout<<"i= "<<i<<"异常了"<<endl;
  247. continue;
  248. }
  249. catch(string& msg){
  250. cout<<"error: "<<msg<<endl;
  251. continue;
  252. }
  253. catch(...)
  254. {
  255. continue;
  256. }
  257. //imshow("pic", img);
  258. //waitKey(1);
  259. cout<<"angle="<<an<<" rootstock_width="<<obj_width<<" time(seconds): "<<((float)t0)/CLOCKS_PER_SEC<<endl;
  260. }
  261. double oa = opa.infer(posinfo);
  262. t = clock() - t;
  263. cout<<"optimal angle: "<<oa<<endl;
  264. cout<<"time(seconds): "<<((float)t)/CLOCKS_PER_SEC<<endl;
  265. cout<<"\n\n";
  266. }
  267. }
  268. destroyAllWindows();
  269. };
  270. void test_optimal_angle_simulate(){
  271. //string folder = "E:\\projects\\grafting_robots\\py_code\\tmp";
  272. string folder = "D:\\grafting_robot\\samples\\tmp";
  273. //string folder = "D:\\grafting_robot\\samples\\rootstock_rotate_part";
  274. //string folder = "D:\\private\\grafting_robot\\samples\\rootstock_rotate_part";
  275. //string folder = "D:\\private\\grafting_robot\\samples\\20211215\\rotate";
  276. namedWindow("pic", CV_WINDOW_NORMAL);
  277. vector<cv::String>filenames;
  278. ConfigParam cp;
  279. test_init_cp(cp);
  280. cp.image_return=true;
  281. cp.image_show=true;
  282. cp.oa_y_flip=false;
  283. cp.oa_clip_y_min=750;
  284. COptimalAnglePart opa(cp,&g_logger);
  285. for(int i = 1;i<=1;++i){
  286. //for(int j = 0;j<=1;++j){
  287. ostringstream ostr;
  288. ostr<<i;
  289. string subfold = ostr.str();
  290. string src_folder = folder+"\\"+subfold;
  291. cv::glob(src_folder, filenames);
  292. opa.reset();
  293. PositionInfo posinfo;
  294. clock_t t,t0;
  295. t = clock();
  296. for(size_t idx=0; idx<filenames.size();++idx){
  297. //cout<<filenames[idx]<<endl;
  298. size_t found0 = filenames[idx].rfind("\\");
  299. size_t found1 = filenames[idx].rfind(".");
  300. int an = stoi(filenames[idx].substr(found0+1, found1-found0));
  301. if(an >=200){
  302. an-=200;
  303. }
  304. Mat img = imread(filenames[idx], cv::IMREAD_GRAYSCALE);
  305. //Rect rect(Point(0,0),Point(img_.cols,cp.oa_clip_y_min));
  306. //Mat img = img_(rect);
  307. ImgInfo* imginfo = mat2imginfo(img);
  308. imginfo->angle = an;
  309. int obj_width=0;
  310. try{
  311. memset(&posinfo,0,sizeof(PositionInfo));
  312. t0 = clock();
  313. obj_width = opa.append(imginfo,posinfo);
  314. t0 = clock() - t0;
  315. imginfo_release(&imginfo);
  316. }
  317. catch(int i)
  318. {
  319. cout<<"i= "<<i<<"异常了"<<endl;
  320. continue;
  321. }
  322. catch(string& msg){
  323. cout<<"error: "<<msg<<endl;
  324. continue;
  325. }
  326. catch(...)
  327. {
  328. continue;
  329. }
  330. //show return images
  331. for (int i =0;i<5;++i){
  332. if (!posinfo.pp_images[i]){continue;}
  333. Mat tmp_img = imginfo2mat(posinfo.pp_images[i]);
  334. imshow("pic", tmp_img);
  335. waitKey(-1);
  336. }
  337. cout<<"angle="<<an<<" rootstock_width="<<obj_width<<" time(seconds): "<<((float)t0)/CLOCKS_PER_SEC<<endl;
  338. }
  339. try{
  340. double oa = opa.infer(posinfo);
  341. t = clock() - t;
  342. cout<<"optimal angle: "<<oa<<endl;
  343. cout<<"time(seconds): "<<((float)t)/CLOCKS_PER_SEC<<endl;
  344. cout<<"\n\n";
  345. }
  346. catch(string& msg){
  347. cout<<"error: "<<msg<<endl;
  348. continue;
  349. }
  350. catch(...)
  351. {
  352. continue;
  353. }
  354. //}
  355. }
  356. destroyAllWindows();
  357. };
  358. void test_optimal_angle_part(){
  359. /*
  360. string data_file = "D:\\private\\grafting_robot\\py_code\\test.txt";
  361. ifstream ifs;
  362. ifs.open(data_file, fstream::out);
  363. std::vector<std::map<int,int>>data;
  364. std::vector<string> answer;
  365. if(ifs.is_open()){
  366. string line;
  367. std::map<int,int>tmp;
  368. while(!ifs.eof()){
  369. getline(ifs, line);
  370. //cout<<line<<endl;
  371. size_t ptmp = line.find("tmp");
  372. if(ptmp!=string::npos){
  373. tmp.clear();
  374. }
  375. else{
  376. size_t pos = line.find(",");
  377. if(pos==string::npos){
  378. data.push_back(tmp);
  379. tmp.clear();
  380. answer.push_back(line);
  381. }
  382. else{
  383. string ang = line.substr(0,pos);
  384. int an = atoi(ang.c_str());
  385. string wid = line.substr(pos+1,string::npos);
  386. int wi = atoi(wid.c_str());
  387. //cout<<ang<<" "<<wid<<endl;
  388. tmp.insert(make_pair<int,int>(an,wi));
  389. }
  390. }
  391. }
  392. ifs.close();
  393. }
  394. //calculate
  395. ConfigParam cp;
  396. test_init_cp(cp);
  397. cp.image_return=true;
  398. cp.image_show=true;
  399. COptimalAnglePart opa(cp);
  400. for(size_t i=0;i< data.size();++i){
  401. cout<<"optimal angle: "<<answer[i]<<endl;
  402. for(size_t j=0;j<5;++j){
  403. int start_angle = j*20;
  404. std::map<int,int> impl_data;
  405. int impl_an = start_angle;
  406. if (impl_an==80 && data[i][impl_an]==277)
  407. {
  408. cout<<"debug"<<endl;
  409. }
  410. while(true){
  411. if((impl_an-start_angle)>110){break;}
  412. if(impl_an>180){break;}
  413. impl_data.insert(make_pair<int,int>(impl_an,data[i][impl_an]));
  414. cout<<impl_an<<"\t"<<data[i][impl_an]<<endl;
  415. impl_an+=20;
  416. }
  417. double a = opa.angle_fit(impl_data);
  418. cout<<"angle: "<<a<<"\n"<<endl;
  419. }
  420. }*/
  421. ConfigParam cp;
  422. test_init_cp(cp);
  423. cp.image_return=true;
  424. cp.image_show=true;
  425. COptimalAnglePart* opa=new COptimalAnglePart(cp);
  426. map<int,int> d;
  427. d.insert(make_pair<int,int>(0,167));
  428. d.insert(make_pair<int,int>(30,126));
  429. d.insert(make_pair<int,int>(60,202));
  430. d.insert(make_pair<int,int>(90,193));
  431. d.insert(make_pair<int,int>(120,121));
  432. double a = opa->angle_fit(d);
  433. cout<<"angle: "<<a<<"\n"<<endl;
  434. };
  435. void test_sc_cut_point()
  436. {
  437. //string src_folder = "E:\\projects\\grafting_robots\\samples\\scion1_part";
  438. //string src_folder = "D:\\private\\grafting_robot\\samples\\scion1_part";
  439. string src_folder = "D:\\grafting_robot\\samples\\scion1_part";
  440. vector<cv::String>filenames;
  441. cv::glob(src_folder, filenames);
  442. ConfigParam cp;
  443. test_init_cp(cp);
  444. /*
  445. //cp.sc_col_th_ratio=0.7;// = 0.7, scion binary image column histogram, threshold ratio for max-value, for
  446. // // detect stem x-range
  447. // cp.sc_row_th_ratio=1.2; // = 1.2, row histogram of stem x-range subimage, stem diameter ratio for detect
  448. // // stem fork position
  449. // cp.sc_stem_x_padding=50; // = 50;
  450. // cp.sc_stem_dia_min=20; //=20,
  451. // cp.sc_stem_dia_max=60; //=60,
  452. // cp.sc_stem_fork_y_min=80;//=80, cut slop length, jump this range for seaching stem fork position
  453. // cp.sc_fork_down=10; // stem fork down X pixels, below this line to calculate r2 ratio
  454. // cp.sc_r2_th=1.05; //threshold for r2 ratio
  455. // cp.sc_r2_window=5; //= 5; the radius for calculate r2 index
  456. // cp.sc_average_window=10;// =10;
  457. //cp.sc_morph_radius=1;//scion, open-operation morph-size, = 1;-->COptimalAngle::imgproc(Mat& img)
  458. // cp.sc_morph_iteration=2; //
  459. //
  460. */
  461. PositionInfo posinfo;
  462. CScionCutPoint scp(cp);
  463. //namedWindow("pic", WINDOW_NORMAL);
  464. ofstream g_logger_ofs;
  465. CGcvLogger g_logger = CGcvLogger(
  466. g_logger_ofs,
  467. CGcvLogger::file_and_terminal,
  468. CGcvLogger::warning,
  469. "./gcv_debug.log");
  470. clock_t t;
  471. for(size_t idx=0; idx<filenames.size();++idx){
  472. //cout<<idx<<"\t"<<filenames[idx]<<endl;
  473. //if(idx<38){continue;}
  474. //string fn = "D:\\private\\grafting_robot\\samples\\scion1_part\\IMG_20210830_153852.jpg";
  475. //Mat img_src = imread(fn, cv::IMREAD_COLOR);
  476. Mat img_src = imread(filenames[idx], cv::IMREAD_COLOR);
  477. Mat img;
  478. resize(img_src,img,Size(img_src.cols/4, img_src.rows/4));
  479. //ImgInfo* imginfo = mat2imginfo(img);
  480. int fold_y = 0;
  481. try{
  482. t = clock();
  483. fold_y = scp.up_point_detect(0,img,posinfo);
  484. t = clock() - t;
  485. }
  486. catch(exception &err){
  487. cout<<err.what()<<endl;
  488. }
  489. catch(const char* msg){
  490. cout<<msg<<endl;
  491. g_logger.ERRORINFO(msg);
  492. g_logger.INFO(filenames[idx]);
  493. }
  494. catch(string msg){
  495. cout<<msg<<endl;
  496. g_logger.ERRORINFO(msg);
  497. g_logger.INFO(filenames[idx]);
  498. }
  499. catch(...){
  500. cout<<"============================================unknown error"<<endl;
  501. cout<<filenames[idx]<<endl<<endl;
  502. }
  503. //cv::line(img,Point(0,fold_y), Point(img.cols,fold_y),Scalar(0,0,255));
  504. //imshow("pic", img);
  505. //waitKey(-1);
  506. cout<<"time(seconds): "<<((float)t)/CLOCKS_PER_SEC<<endl;
  507. cout<<"\n";
  508. }
  509. };
  510. void test_sc_cut_point_simulate()
  511. {
  512. //string src_folder = "E:\\projects\\grafting_robots\\samples\\scion1_part";
  513. //string src_folder = "D:\\private\\grafting_robot\\samples\\scion1_part";
  514. //string src_folder = "D:\\grafting_robot\\samples\\scion1_simulate";
  515. string src_folder = "D:\\private\\grafting_robot\\samples\\20211222\\scion";
  516. vector<cv::String>filenames;
  517. cv::glob(src_folder, filenames);
  518. ConfigParam cp;
  519. test_init_cp(cp);
  520. cp.image_return=true;
  521. cp.image_show=true;
  522. cp.rs_y_flip=false;
  523. cp.self_camera=false;
  524. PositionInfo posinfo;
  525. memset(&posinfo,0,sizeof(PositionInfo));
  526. CScionCutPoint scp(cp,&g_logger);
  527. //namedWindow("pic", WINDOW_NORMAL);
  528. /*ofstream g_logger_ofs;
  529. Logger g_logger = Logger(
  530. g_logger_ofs,
  531. Logger::file_and_terminal,
  532. Logger::warning,
  533. "./gcv_debug.log");*/
  534. clock_t t;
  535. for(size_t idx=0; idx<filenames.size();++idx){
  536. //cout<<idx<<"\t"<<filenames[idx]<<endl;
  537. //if(idx<38){continue;}
  538. //string fn = "D:\\private\\grafting_robot\\samples\\scion1_part\\IMG_20210830_153852.jpg";
  539. //Mat img_src = imread(fn, cv::IMREAD_COLOR);
  540. cout<<idx<<"\t"<<filenames[idx]<<endl;
  541. //if(filenames[idx].find("0-4-504")==string::npos){continue;}
  542. Mat img = imread(filenames[idx], cv::IMREAD_COLOR);
  543. int fold_y = 0;
  544. try{
  545. t = clock();
  546. fold_y = scp.up_point_detect(0,img,posinfo);
  547. t = clock() - t;
  548. }
  549. catch(exception &err){
  550. cout<<err.what()<<endl;
  551. }
  552. catch(const char* msg){
  553. cout<<msg<<endl;
  554. g_logger.ERRORINFO(msg);
  555. g_logger.INFO(filenames[idx]);
  556. }
  557. catch(string msg){
  558. cout<<msg<<endl;
  559. g_logger.ERRORINFO(msg);
  560. g_logger.INFO(filenames[idx]);
  561. }
  562. catch(...){
  563. cout<<"============================================unknown error"<<endl;
  564. cout<<filenames[idx]<<endl<<endl;
  565. }
  566. //show return images
  567. for (int i =0;i<5;++i){
  568. if (!posinfo.pp_images[i]){continue;}
  569. Mat tmp_img = imginfo2mat(posinfo.pp_images[i]);
  570. imshow("pic", tmp_img);
  571. waitKey(-1);
  572. }
  573. //cv::line(img,Point(0,fold_y), Point(img.cols,fold_y),Scalar(0,0,255));
  574. //imshow("pic", img);
  575. //waitKey(-1);
  576. cout<<"time(seconds): "<<((float)t)/CLOCKS_PER_SEC<<endl;
  577. cout<<"\n";
  578. }
  579. };
  580. void test_rs_cut_point(){
  581. ConfigParam cp;
  582. test_init_cp(cp);
  583. namedWindow("pic", CV_WINDOW_NORMAL);
  584. CRootStockCutPoint rscp(cp);
  585. ofstream g_logger_ofs;
  586. CGcvLogger g_logger = CGcvLogger(
  587. g_logger_ofs,
  588. CGcvLogger::file_and_terminal,
  589. CGcvLogger::warning,
  590. "./gcv_debug.log");
  591. clock_t t;
  592. for(int i=4;i<193;++i){
  593. //if (/*i !=11 &&*/ D:\private\grafting_robot\samples\rs_cut_simulate
  594. // i !=32 &&
  595. // i !=48 &&
  596. // i != 49 &&
  597. // i != 53 &&
  598. // i != 103 &&
  599. // i != 104 &&
  600. // i != 106 &&
  601. // i != 110 &&
  602. // i != 125 &&
  603. // i !=182 &&
  604. // i != 187 &&
  605. // i != 191)
  606. //{
  607. // continue;
  608. //}
  609. stringstream buff;
  610. buff<<setw(3) << setfill('0') << i;
  611. cout<<buff.str()<<endl;
  612. //string img_file= "E:/projects/grafting_robots/samples/rootstlock_pumpkin/"+buff.str()+"/6.jpg";
  613. //string img_file= "D:/private/grafting_robot/samples/rootstlock_pumpkin/"+buff.str()+"/6.jpg";
  614. string img_file= "D:/grafting_robot/samples/rootstlock_pumpkin/"+buff.str()+"/6.jpg";
  615. Mat img = imread(img_file,CV_LOAD_IMAGE_COLOR );
  616. //ImgInfo* imginfo = mat2imginfo(img);
  617. PositionInfo pinfo;
  618. try{
  619. t = clock();
  620. int fold_y = rscp.up_point_detect(0,img, pinfo);
  621. t = clock() - t;
  622. }
  623. catch(exception &err){
  624. cout<<err.what()<<endl;
  625. }
  626. catch(string msg){
  627. cout<<msg<<endl;
  628. cout<<img_file<<endl<<endl;
  629. g_logger.ERRORINFO(msg);
  630. g_logger.INFO(buff.str());
  631. }
  632. catch(...){
  633. cout<<"============================================unknown error"<<endl;
  634. cout<<img_file<<endl<<endl;
  635. }
  636. //cv::line(img,Point(0,fold_y), Point(img.cols,fold_y),Scalar(0,0,255));
  637. imshow("pic", img);
  638. waitKey(1);
  639. cout<<"time(seconds): "<<((float)t)/CLOCKS_PER_SEC<<endl;
  640. cout<<"\n\n";
  641. }
  642. };
  643. void test_rs_cut_point_simulate(){
  644. ConfigParam cp;
  645. test_init_cp(cp);
  646. cp.image_show=true;
  647. cp.image_return = true;
  648. namedWindow("pic", CV_WINDOW_NORMAL);
  649. CRootStockCutPoint rscp(cp,&g_logger);
  650. /*ofstream g_logger_ofs;
  651. Logger g_logger = Logger(
  652. g_logger_ofs,
  653. Logger::file_and_terminal,
  654. Logger::warning,
  655. "./gcv_debug.log");*/
  656. //string src_folder = "D:\\private\\grafting_robot\\samples\\rs_cut_simulate";
  657. //string src_folder = "D:\\private\\grafting_robot\\samples\\rootstock_hold_down";
  658. //string src_folder = "D:\\grafting_robot\\samples\\rootstock_hold_down";
  659. string src_folder = "D:\\private\\grafting_robot\\samples\\20211215\\rootstock";
  660. vector<cv::String>filenames;
  661. cv::glob(src_folder, filenames);
  662. clock_t t;
  663. for(int i=0;i<filenames.size();++i){
  664. string img_file= filenames[i];
  665. Mat img = imread(img_file,CV_LOAD_IMAGE_COLOR );
  666. //ImgInfo* imginfo = mat2imginfo(img);
  667. PositionInfo pinfo;
  668. memset(&pinfo,0,sizeof(PositionInfo));
  669. try{
  670. t = clock();
  671. int fold_y = rscp.up_point_detect(0,img, pinfo);
  672. t = clock() - t;
  673. }
  674. catch(exception &err){
  675. cout<<err.what()<<endl;
  676. }
  677. catch(string msg){
  678. cout<<msg<<endl;
  679. cout<<img_file<<endl<<endl;
  680. g_logger.ERRORINFO(msg);
  681. g_logger.INFO(img_file);
  682. }
  683. catch(...){
  684. cout<<"============================================unknown error"<<endl;
  685. cout<<img_file<<endl<<endl;
  686. }
  687. //cv::line(img,Point(0,fold_y), Point(img.cols,fold_y),Scalar(0,0,255));
  688. imshow("pic", img);
  689. waitKey(1);
  690. //show return images
  691. for (int i =0;i<5;++i){
  692. if (!pinfo.pp_images[i]){continue;}
  693. Mat tmp_img = imginfo2mat(pinfo.pp_images[i]);
  694. imshow("pic", tmp_img);
  695. waitKey(-1);
  696. }
  697. cout<<"time(seconds): "<<((float)t)/CLOCKS_PER_SEC<<endl;
  698. cout<<"\n\n";
  699. }
  700. };
  701. //void test_rs_cp_reid()
  702. //{
  703. // ConfigParam cp;
  704. // test_init_cp(cp);
  705. // cp.image_show=false;
  706. // cp.image_return = true;
  707. // cp.self_camera=false;
  708. // cp.oa_y_flip=false;
  709. //
  710. // namedWindow("pic", CV_WINDOW_NORMAL);
  711. // CRootStockCutPoint rscp(cp,&g_logger);
  712. //
  713. // //string src_folder = "D:\\private\\grafting_robot\\samples\\rs_cut_simulate";
  714. // //string src_folder = "D:\\private\\grafting_robot\\samples\\rootstock_hold_down";
  715. // //string src_folder = "D:\\grafting_robot\\samples\\rootstock_hold_down";
  716. // string src_folder = "D:\\private\\grafting_robot\\samples\\20211215\\reid";
  717. // vector<cv::String>filenames;
  718. // cv::glob(src_folder, filenames);
  719. //
  720. // clock_t t;
  721. // for(int i=0;i<filenames.size();++i){
  722. //
  723. // string img_file= filenames[i];
  724. //
  725. // Mat img = imread(img_file,CV_LOAD_IMAGE_COLOR);
  726. // //img = img(Rect(0,0,img.cols,(int)(img.rows/2)));
  727. // imshow("pic", img);
  728. // waitKey(-1);
  729. //
  730. // //ImgInfo* imginfo = mat2imginfo(img);
  731. // PositionInfo pinfo;
  732. // memset(&pinfo,0,sizeof(PositionInfo));
  733. // try{
  734. // t = clock();
  735. // int fold_y = rscp.up_point_reid(0,img,100.0, pinfo);
  736. // t = clock() - t;
  737. // }
  738. // catch(exception &err){
  739. // cout<<err.what()<<endl;
  740. // }
  741. // catch(string msg){
  742. // cout<<msg<<endl;
  743. // cout<<img_file<<endl<<endl;
  744. // g_logger.ERRORINFO(msg);
  745. // g_logger.INFO(img_file);
  746. // }
  747. // catch(...){
  748. // cout<<"============================================unknown error"<<endl;
  749. // cout<<img_file<<endl<<endl;
  750. // }
  751. // //cv::line(img,Point(0,fold_y), Point(img.cols,fold_y),Scalar(0,0,255));
  752. //
  753. // imshow("pic", img);
  754. // waitKey(1);
  755. //
  756. // //show return images
  757. // for (int i =0;i<5;++i){
  758. // if (!pinfo.pp_images[i]){continue;}
  759. // Mat tmp_img = imginfo2mat(pinfo.pp_images[i]);
  760. // imshow("pic", tmp_img);
  761. // waitKey(-1);
  762. // }
  763. // cout<<"time(seconds): "<<((float)t)/CLOCKS_PER_SEC<<endl;
  764. //
  765. // cout<<"\n\n";
  766. //
  767. // }
  768. //}
  769. void test_api_scion(){
  770. // 0 ConfigParam cp;
  771. ConfigParam cp;
  772. //0
  773. //char* lpath="D:\\grafting_robot\\cpp_code\\logs\\gcv.log";
  774. //char* lpath = "D:\\private\\grafting_robot\\logs\\gcv.log";
  775. //cv_set_logpath(lpath);
  776. //cv_set_loglevel(0);
  777. //cout<<"test"<<endl;
  778. // 1 get version
  779. //test_init_cp(cp);
  780. char* ver = new char[10];
  781. get_version(ver);
  782. cout<<ver<<endl;
  783. delete [] ver;
  784. ver=0;
  785. //2 cv_set_param
  786. //cp.image_show=false;
  787. //cv_set_param(cp);
  788. //cv_save_param(0);
  789. //3 set log
  790. //cv_set_logpath("D:\\logs");
  791. //cv_set_loglevel(0);
  792. //3 cv_get_param();
  793. //ConfigParam cp_tmp;
  794. //cv_get_param(cp_tmp);
  795. //4 void cv_get_conf_file
  796. //char* conf_file_ret = new char[128];
  797. //cv_get_conf_file(conf_file_ret);
  798. //cout<<conf_file_ret<<endl;
  799. //delete [] conf_file_ret;
  800. //conf_file_ret=0;
  801. //5 cv_init()
  802. //char *conf_file = "D:\\private\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  803. //char *conf_file = "D:\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  804. char *conf_file = "E:\\projects\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  805. int rst = cv_init(conf_file);
  806. ConfigParam cp_ret;
  807. cv_get_param(cp_ret);
  808. cv_set_loglevel(0);
  809. cv_init_image_saver();
  810. //return;
  811. //
  812. //string work_folder = "D:\\private\\grafting_robot\\samples\\20211215\\scion";
  813. string work_folder = "E:\\projects\\grafting_robot\\samples\\20220115\\scion";
  814. vector<cv::String>filenames;
  815. cv::glob(work_folder, filenames);
  816. for(size_t i=0;i<filenames.size();++i){
  817. string fname = filenames[i];
  818. PositionInfo posinfo;
  819. Mat img = imread(fname, cv::IMREAD_GRAYSCALE);
  820. if(img.empty()){continue;}
  821. image_set_top(img,20,8);
  822. ImgInfo* imginfo = mat2imginfo(img);
  823. int rst = sc_cut_point(imginfo,posinfo);
  824. for (int i =0;i<5;++i){
  825. if (!posinfo.pp_images[i]){continue;}
  826. Mat tmp_img = imginfo2mat(posinfo.pp_images[i]);
  827. imshow("pic", tmp_img);
  828. waitKey(-1);
  829. }
  830. }
  831. cv_release();
  832. }
  833. void test_sc_batch(){
  834. // 0 version;
  835. char* ver = new char[10];
  836. get_version(ver);
  837. cout<<ver<<endl;
  838. delete [] ver;
  839. ver=0;
  840. //2 cv_init()
  841. //char *conf_file = "D:\\private\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  842. //char *conf_file = "D:\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  843. char *conf_file = "E:\\projects\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  844. int rst = cv_init(conf_file);
  845. ConfigParam cp_ret;
  846. cv_get_param(cp_ret);
  847. cv_set_loglevel(0);
  848. cv_init_image_saver();
  849. //cp_ret.image_show = true;
  850. //cv_set_param(cp_ret);
  851. //namedWindow("pic", CV_WINDOW_NORMAL);
  852. string work_folder = "E:\\projects\\grafting_robot\\samples\\20220223\\scion";
  853. //string work_folder = "D:\\private\\grafting_robot\\samples\\20220220\\scion";
  854. vector<cv::String>filenames;
  855. cv::glob(work_folder, filenames);
  856. for(size_t idx=0; idx<filenames.size();++idx){
  857. /*stringstream buff;
  858. buff<<work_folder<<"\\"<<i;
  859. string batch_folder = buff.str();*/
  860. PositionInfo posinfo;
  861. string filename = filenames[idx];
  862. if(filename.find("rst_")!=string::npos){
  863. continue;
  864. }
  865. cout<<idx<<"\t"<<filename<<endl;
  866. Mat img = imread(filename, cv::IMREAD_GRAYSCALE);
  867. if(img.empty()){continue;}
  868. image_set_top(img,20,8);
  869. ImgInfo* imginfo = mat2imginfo(img);
  870. try{
  871. //if(filename.find("1947_3.jpg")!=string::npos){
  872. // int ooo=0;
  873. // /*cp_ret.image_show=true;
  874. // cv_set_param(cp_ret);*/
  875. //}
  876. //else{
  877. // continue;
  878. //}
  879. int rst = sc_cut_point(imginfo, posinfo);
  880. imginfo_release(&imginfo);
  881. if(rst){
  882. cout<<"error"<<endl;
  883. continue;
  884. }
  885. for (int i =0;i<5;++i){
  886. if (!posinfo.pp_images[i]){continue;}
  887. Mat tmp_img = imginfo2mat(posinfo.pp_images[i]);
  888. /*imshow("pic", tmp_img);
  889. waitKey(-1);*/
  890. stringstream bbu;
  891. bbu<<filename<<".rst_"<<i<<".jpg";
  892. string fnn = bbu.str();
  893. cv::imwrite(fnn,tmp_img);
  894. }
  895. }
  896. catch(...){
  897. std::cout<<"some error ..."<<endl;
  898. }
  899. }
  900. cv_release();
  901. }
  902. void test_rs_batch(){
  903. // 0 version;
  904. char* ver = new char[10];
  905. get_version(ver);
  906. cout<<ver<<endl;
  907. delete [] ver;
  908. ver=0;
  909. //2 cv_init()
  910. //char *conf_file = "D:\\private\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  911. //char *conf_file = "D:\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  912. char *conf_file = "E:\\projects\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  913. int rst = cv_init(conf_file);
  914. ConfigParam cp_ret;
  915. cv_get_param(cp_ret);
  916. cv_set_loglevel(0);
  917. cv_init_image_saver();
  918. //cp_ret.image_show = true;
  919. //cv_set_param(cp_ret);
  920. //namedWindow("pic", CV_WINDOW_NORMAL);
  921. string work_folder = "E:\\projects\\grafting_robot\\samples\\20220620\\rootstock";
  922. //string work_folder = "D:\\private\\grafting_robot\\samples\\20220220\\rootstock";
  923. vector<cv::String>filenames;
  924. cv::glob(work_folder, filenames);
  925. int cnter=0;
  926. for(size_t idx=0; idx<filenames.size();++idx){
  927. string rs_filename = filenames[idx];
  928. if(rs_filename.find("rst_")!=string::npos){
  929. continue;
  930. }
  931. //if(rs_filename.find("\\379.bmp")==string::npos){continue;}
  932. cout<<idx<<"\t"<<rs_filename<<endl;
  933. PositionInfo posinfo;
  934. Mat img = imread(rs_filename,cv::IMREAD_GRAYSCALE);
  935. image_set_top(img,20,8);
  936. ImgInfo* rs_imginfo = mat2imginfo(img);
  937. memset(&posinfo,0,sizeof(PositionInfo));
  938. try{
  939. /*if(cnter==4){
  940. int test_tmp=0;
  941. }*/
  942. int fold_y = rs_cut_point(rs_imginfo, posinfo);
  943. imginfo_release(&rs_imginfo);
  944. for (int i =0;i<5;++i){
  945. if (!posinfo.pp_images[i]){continue;}
  946. Mat tmp_img = imginfo2mat(posinfo.pp_images[i]);
  947. /*imshow("pic", tmp_img);
  948. waitKey(-1);*/
  949. stringstream bbu;
  950. bbu<<rs_filename<<".rst_"<<i<<".jpg";
  951. string fnn = bbu.str();
  952. cv::imwrite(fnn,tmp_img);
  953. }
  954. }
  955. catch(exception &err){
  956. cout<<err.what()<<endl;
  957. }
  958. catch(string msg){
  959. cout<<msg<<endl;
  960. cout<<rs_filename<<endl<<endl;
  961. //g_logger.ERRORINFO(msg);
  962. //g_logger.INFO(img_file);
  963. }
  964. catch(...){
  965. cout<<"============================================unknown error"<<endl;
  966. cout<<rs_filename<<endl<<endl;
  967. }
  968. cnter+=1;
  969. }
  970. cv_release();
  971. }
  972. void test_rs_batch_camera(){
  973. // 0 version;
  974. char* ver = new char[10];
  975. get_version(ver);
  976. cout<<ver<<endl;
  977. delete [] ver;
  978. ver=0;
  979. //2 cv_init()
  980. //char *conf_file = "D:\\private\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  981. //char *conf_file = "D:\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  982. char *conf_file = "E:\\projects\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  983. int rst = cv_init(conf_file);
  984. ConfigParam cp_ret;
  985. cv_get_param(cp_ret);
  986. cv_set_loglevel(0);
  987. cv_init_image_saver();
  988. //cp_ret.image_show = true;
  989. //cv_set_param(cp_ret);
  990. //namedWindow("pic", CV_WINDOW_NORMAL);
  991. string work_folder = "E:\\projects\\grafting_robot\\samples\\20220119\\rootstock_cut_compare";
  992. //string work_folder = "D:\\private\\grafting_robot\\samples\\20220104\\batch";
  993. vector<string> sub_path;
  994. sub_path.push_back("20220119133425");
  995. sub_path.push_back("20220119134921");
  996. sub_path.push_back("20220119135818");
  997. sub_path.push_back("20220119140720");
  998. sub_path.push_back("20220119141511");
  999. sub_path.push_back("20220119142616");
  1000. sub_path.push_back("20220119143048");
  1001. //vector<cv::String>filenames;
  1002. //cv::glob(work_folder, filenames);
  1003. vector<string> fileidx;
  1004. fileidx.push_back("3.bmp");
  1005. fileidx.push_back("11.bmp");
  1006. fileidx.push_back("19.bmp");
  1007. fileidx.push_back("27.bmp");
  1008. fileidx.push_back("35.bmp");
  1009. fileidx.push_back("43.bmp");
  1010. fileidx.push_back("51.bmp");
  1011. fileidx.push_back("59.bmp");
  1012. fileidx.push_back("67.bmp");
  1013. fileidx.push_back("75.bmp");
  1014. fileidx.push_back("83.bmp");
  1015. fileidx.push_back("91.bmp");
  1016. fileidx.push_back("99.bmp");
  1017. fileidx.push_back("107.bmp");
  1018. fileidx.push_back("115.bmp");
  1019. fileidx.push_back("123.bmp");
  1020. for(size_t idx=0; idx<sub_path.size();++idx){
  1021. string subp = sub_path[idx];
  1022. for(size_t i=0;i<fileidx.size();++i){
  1023. string rs_filename = work_folder+"\\"+subp+ "\\"+fileidx[i];
  1024. PositionInfo posinfo;
  1025. Mat img = imread(rs_filename,cv::IMREAD_GRAYSCALE);
  1026. if(img.empty()){continue;}
  1027. ImgInfo* rs_imginfo = mat2imginfo(img);
  1028. memset(&posinfo,0,sizeof(PositionInfo));
  1029. try{
  1030. int fold_y = rs_cut_point(rs_imginfo, posinfo);
  1031. imginfo_release(&rs_imginfo);
  1032. for (int i =0;i<5;++i){
  1033. if (!posinfo.pp_images[i]){continue;}
  1034. Mat tmp_img = imginfo2mat(posinfo.pp_images[i]);
  1035. /*imshow("pic", tmp_img);
  1036. waitKey(-1);*/
  1037. stringstream bbu;
  1038. bbu<<rs_filename<<".rst_"<<i<<".jpg";
  1039. string fnn = bbu.str();
  1040. cv::imwrite(fnn,tmp_img);
  1041. }
  1042. }
  1043. catch(exception &err){
  1044. cout<<err.what()<<endl;
  1045. }
  1046. catch(string msg){
  1047. cout<<msg<<endl;
  1048. cout<<rs_filename<<endl<<endl;
  1049. //g_logger.ERRORINFO(msg);
  1050. //g_logger.INFO(img_file);
  1051. }
  1052. catch(...){
  1053. cout<<"============================================unknown error"<<endl;
  1054. cout<<rs_filename<<endl<<endl;
  1055. }
  1056. }
  1057. }
  1058. cv_release();
  1059. }
  1060. void test_oa_batch(){
  1061. // 0 version;
  1062. char* ver = new char[10];
  1063. get_version(ver);
  1064. cout<<ver<<endl;
  1065. delete [] ver;
  1066. ver=0;
  1067. //2 cv_init()
  1068. //char *conf_file = "D:\\private\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  1069. //char *conf_file = "D:\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  1070. char *conf_file = "E:\\projects\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  1071. int rst = cv_init(conf_file);
  1072. ConfigParam cp_ret;
  1073. cv_get_param(cp_ret);
  1074. cv_set_loglevel(0);
  1075. cv_init_image_saver();
  1076. //cp_ret.image_show=true;
  1077. //cv_set_param(cp_ret);
  1078. //namedWindow("pic", CV_WINDOW_NORMAL);
  1079. string work_folder = "E:\\projects\\grafting_robot\\samples\\20220223\\rotate";
  1080. //string work_folder = "D:\\private\\grafting_robot\\samples\\20220220\\rotate";
  1081. vector<cv::String>filenames;
  1082. cv::glob(work_folder, filenames);
  1083. for(size_t idx=0; idx<filenames.size();++idx){
  1084. string filename = filenames[idx];
  1085. if(filename.find(".rst_")!=string::npos){continue;}
  1086. //if(filename.find("\\249.bmp")==string::npos){continue;}
  1087. PositionInfo posinfo;
  1088. Mat img = imread(filename, cv::IMREAD_GRAYSCALE);
  1089. if(img.empty()){continue;}
  1090. image_set_top(img,20,8);
  1091. ImgInfo* imginfo = mat2imginfo(img);
  1092. //imginfo->angle = j*30;
  1093. try{
  1094. rs_oa_init();
  1095. rs_oa_append(imginfo, posinfo);
  1096. for (int i =0;i<5;++i){
  1097. if (!posinfo.pp_images[i]){continue;}
  1098. Mat tmp_img = imginfo2mat(posinfo.pp_images[i]);
  1099. /*imshow("pic", tmp_img);
  1100. waitKey(-1);*/
  1101. stringstream bbu;
  1102. bbu<<filename<<".rst_"<<i<<".jpg";
  1103. string fnn = bbu.str();
  1104. cv::imwrite(fnn,tmp_img);
  1105. }
  1106. }
  1107. catch(...){
  1108. std::cout<<"some error ..."<<endl;
  1109. }
  1110. imginfo_release(&imginfo);
  1111. }
  1112. cv_release();
  1113. }
  1114. void test_api_batch(){
  1115. // 0 version;
  1116. char* ver = new char[10];
  1117. get_version(ver);
  1118. cout<<ver<<endl;
  1119. delete [] ver;
  1120. ver=0;
  1121. //2 cv_init()
  1122. //char *conf_file = "D:\\private\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  1123. //char *conf_file = "D:\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  1124. char *conf_file = "E:\\projects\\grafting_robot\\cpp_code\\demo\\demo\\gcv_conf.yml";
  1125. int rst = cv_init(conf_file);
  1126. ConfigParam cp_ret;
  1127. cv_get_param(cp_ret);
  1128. cv_set_loglevel(0);
  1129. cv_init_image_saver();
  1130. namedWindow("pic", CV_WINDOW_NORMAL);
  1131. string work_folder = "E:\\projects\\grafting_robot\\samples\\20220105\\batch";
  1132. //string work_folder = "D:\\private\\grafting_robot\\samples\\20211215\\scion";
  1133. //string work_folder = "D:\\grafting_robot\\samples\\20220102\\batch";
  1134. for(size_t i=1;i<19;++i){
  1135. stringstream buff;
  1136. buff<<work_folder<<"\\"<<i;
  1137. //buff<<work_folder<<"\\"<<"0-4-489.bmp";
  1138. string batch_folder = buff.str();
  1139. PositionInfo posinfo;
  1140. ///////////////////////////////////////////////////////
  1141. // 1 optimal angle
  1142. /*rs_oa_init();
  1143. for(size_t j=0;j<5;++j){
  1144. stringstream bf;
  1145. bf<<batch_folder<<"\\"<<2*j+1<<".bmp";
  1146. string filename = bf.str();
  1147. Mat img = imread(filename, cv::IMREAD_GRAYSCALE);
  1148. if(img.empty()){continue;}
  1149. image_set_top(img,20,8);
  1150. ImgInfo* imginfo = mat2imginfo(img);
  1151. imginfo->angle = j*30;
  1152. try{
  1153. rs_oa_append(imginfo, posinfo);
  1154. }
  1155. catch(...){
  1156. std::cout<<"some error ..."<<endl;
  1157. }
  1158. std::cout<<"angle="<<imginfo->angle<<" rootstock_width="<<posinfo.rs_oa_width<<endl;
  1159. imginfo_release(&imginfo);
  1160. for (int i =0;i<5;++i){
  1161. if (!posinfo.pp_images[i]){continue;}
  1162. Mat tmp_img = imginfo2mat(posinfo.pp_images[i]);
  1163. imshow("pic", tmp_img);
  1164. waitKey(-1);
  1165. }
  1166. }
  1167. rs_oa_get_result(posinfo);
  1168. cout<<"optimal angle="<<posinfo.rs_oa<<endl;
  1169. double opt_angle = posinfo.rs_oa;*/
  1170. //////////////////////////////////////////////////////////////////
  1171. // 2 rootstock cut point
  1172. //if(!(i==4 || i==6 ||i==8 || i==12 || i==15) ){continue;}
  1173. //if(i!=12){continue;}
  1174. string rs_filename = batch_folder+"\\11.bmp";
  1175. /*if (i==15){
  1176. rs_filename = batch_folder+"\\11.jpg";
  1177. }*/
  1178. Mat img = imread(rs_filename,cv::IMREAD_GRAYSCALE);
  1179. image_set_top(img,20,8);
  1180. ImgInfo* rs_imginfo = mat2imginfo(img);
  1181. clock_t t;
  1182. memset(&posinfo,0,sizeof(PositionInfo));
  1183. try{
  1184. t = clock();
  1185. int fold_y = rs_cut_point(rs_imginfo, posinfo);
  1186. t = clock() - t;
  1187. imginfo_release(&rs_imginfo);
  1188. }
  1189. catch(exception &err){
  1190. cout<<err.what()<<endl;
  1191. }
  1192. catch(string msg){
  1193. cout<<msg<<endl;
  1194. cout<<rs_filename<<endl<<endl;
  1195. //g_logger.ERRORINFO(msg);
  1196. //g_logger.INFO(img_file);
  1197. }
  1198. catch(...){
  1199. cout<<"============================================unknown error"<<endl;
  1200. cout<<rs_filename<<endl<<endl;
  1201. }
  1202. //show return images
  1203. /*stringstream tu;
  1204. tu<<batch_folder<<"\\rst_"<<int(opt_angle)<<"_"<<11<<".jpg";
  1205. string tuu = tu.str();
  1206. cv::imwrite(tuu,img);*/
  1207. /*for (int i =0;i<5;++i){
  1208. if (!posinfo.pp_images[i]){continue;}
  1209. Mat tmp_img = imginfo2mat(posinfo.pp_images[i]);
  1210. imshow("pic", tmp_img);
  1211. waitKey(-1);
  1212. stringstream bbu;
  1213. bbu<<batch_folder<<"\\rst_"<<i<<".jpg";
  1214. string fnn = bbu.str();
  1215. cv::imwrite(fnn,tmp_img);
  1216. } */
  1217. cout<<"time(seconds): "<<((float)t)/CLOCKS_PER_SEC<<endl;
  1218. cout<<"\n\n";
  1219. }
  1220. cv_release();
  1221. }
  1222. void drawline_rs()
  1223. {
  1224. string rs_filename = "D:\\grafting_robot\\samples\\20220108\\rootstock\\27.jpg";
  1225. Mat img = imread(rs_filename,cv::IMREAD_GRAYSCALE);
  1226. double angle = -40.0 * 3.1415926 /180.0;
  1227. //angle =tan(angle );
  1228. int x0=259;
  1229. int y0 = 461;
  1230. int b = x0-y0;
  1231. Point p0=Point(x0,y0);
  1232. int x1 = img.cols-1;
  1233. double dx=x1-x0;
  1234. double dy = tan(angle) * dx;
  1235. if(dy<0){dy = -dy;}
  1236. int y1 = y0+(int)(dy+0.5);
  1237. Point p1 = Point(x1,y1);
  1238. int x2=0;
  1239. int y2 = x0*tan(angle )+y0;
  1240. Point p2 = Point(x2,y2);
  1241. line(img,p0,p1,Scalar(255,255,255));
  1242. line(img,p2,p1,Scalar(255,255,255));
  1243. string fnn = "./tmp27.bmp";
  1244. cv::imwrite(fnn,img);
  1245. }
  1246. void drawline_sc()
  1247. {
  1248. string rs_filename = "D:\\grafting_robot\\samples\\20220108\\rootstock\\26.jpg";
  1249. Mat img = imread(rs_filename,cv::IMREAD_GRAYSCALE);
  1250. double angle = -40.0 * 3.1415926 /180.0;
  1251. //angle =tan(angle );
  1252. int x0=395;
  1253. int y0 = 340;
  1254. int b = x0-y0;
  1255. Point p0=Point(x0,y0);
  1256. int x1 = img.cols-1;
  1257. double dx=x1-x0;
  1258. double dy = tan(angle) * dx;
  1259. if(dy<0){dy = -dy;}
  1260. int y1 = y0+(int)(dy+0.5);
  1261. Point p1 = Point(x1,y1);
  1262. int x2=0;
  1263. int y2 = x0*tan(angle )+y0;
  1264. Point p2 = Point(x2,y2);
  1265. line(img,p0,p1,Scalar(255,255,255));
  1266. line(img,p2,p1,Scalar(255,255,255));
  1267. string fnn = "./tmp26.bmp";
  1268. cv::imwrite(fnn,img);
  1269. }
  1270. void drawline_dist()
  1271. {
  1272. double x0=260.0;
  1273. double y0=292.0;
  1274. double angle = -45.0 * 3.1415926 /180.0;
  1275. double k=tan(angle );
  1276. double b = -k*260.0-435.0;
  1277. double yy = -(k*x0 +b);
  1278. double dy = yy-y0;
  1279. double r = 5.8479999999999997e-002;
  1280. double dyy = dy *r;
  1281. cout<<dy<<"\t"<<dyy;
  1282. }
  1283. int _tmain(int argc, _TCHAR* argv[])
  1284. {
  1285. //test_imginfo2mat();
  1286. //test_camconfig_write();
  1287. //test_camconfig_read();
  1288. //test_anglefit();
  1289. //test_optimal_angle();
  1290. //test_optimal_angle_simulate();
  1291. //test_optimal_angle_part();
  1292. //test_sc_cut_point();
  1293. //test_sc_cut_point_simulate();
  1294. //test_rs_cut_point();
  1295. //test_rs_cut_point_simulate();
  1296. //test_rs_cp_reid();
  1297. /////////////////////////////
  1298. //api test
  1299. //test_api_scion();
  1300. //test_sc_batch();
  1301. test_rs_batch();
  1302. //test_rs_batch_camera();
  1303. //test_oa_batch();
  1304. //test_api_batch();
  1305. //drawline_rs();
  1306. //drawline_sc();
  1307. //drawline_dist();
  1308. //system("pause");
  1309. return 0;
  1310. }