| 2354 | } |
| 2355 | |
| 2356 | void setLabel(cv::Mat& im, const std::string label, const cv::Point& org) { |
| 2357 | int fontface = cv::FONT_HERSHEY_SIMPLEX; |
| 2358 | double scale = 0.5; |
| 2359 | int thickness = 1; |
| 2360 | int baseline = 0; |
| 2361 | |
| 2362 | cv::Size text = cv::getTextSize(label, fontface, scale, thickness, &baseline); |
| 2363 | rectangle(im, org + cv::Point(0, baseline), org + cv::Point(text.width, -text.height), CV_RGB(0,0,0), CV_FILLED); |
| 2364 | putText(im, label, org, fontface, scale, CV_RGB(255,255,255), thickness, 20); |
| 2365 | } |
| 2366 | |
| 2367 | template<typename Dtype> |
| 2368 | void DataTransformer<Dtype>::visualize(cv::Mat& img, MetaData meta, AugmentSelection as) { |