|
|
@ -1120,6 +1120,7 @@ extern "C" void draw_train_loss(char *windows_name, mat_cv* img_src, int img_siz |
|
|
|
if (k == 's' || current_batch == (max_batches - 1) || (current_batch / 100 > old_batch / 100)) { |
|
|
|
if (k == 's' || current_batch == (max_batches - 1) || (current_batch / 100 > old_batch / 100)) { |
|
|
|
old_batch = current_batch; |
|
|
|
old_batch = current_batch; |
|
|
|
save_mat_png(img, "chart.png"); |
|
|
|
save_mat_png(img, "chart.png"); |
|
|
|
|
|
|
|
save_mat_png(img, windows_name); |
|
|
|
cv::putText(img, "- Saved", cv::Point(260, img_size - 10), cv::FONT_HERSHEY_COMPLEX_SMALL, 0.7, CV_RGB(255, 0, 0), 1, CV_AA); |
|
|
|
cv::putText(img, "- Saved", cv::Point(260, img_size - 10), cv::FONT_HERSHEY_COMPLEX_SMALL, 0.7, CV_RGB(255, 0, 0), 1, CV_AA); |
|
|
|
} |
|
|
|
} |
|
|
|
else |
|
|
|
else |
|
|
|