|
@@ -351,7 +351,7 @@ int CTeaSort::detect(
|
|
|
}
|
|
|
}
|
|
|
//ÅÄÕÕÎÞÃç, ·µ»ØÊ¶±ð½á¹û-1
|
|
|
- if (valid_cnt != 2) { return -1; }
|
|
|
+ if (valid_cnt == 0) { return -1; }
|
|
|
return 0;
|
|
|
}
|
|
|
|
|
@@ -453,8 +453,8 @@ int CTeaSort::load_data(
|
|
|
if (imginfo) {
|
|
|
if (m_pLogger) {
|
|
|
stringstream buff;
|
|
|
- buff << m_imgId << m_dtype_str << "image, width=" << imginfo->width
|
|
|
- << "\theight=" << imginfo->height;
|
|
|
+ buff << "raw image stream: " << m_imgId << m_dtype_str << "image, width=" << imginfo->width
|
|
|
+ << "\theight=" << imginfo->height << "\tchannels=" << imginfo->channel;
|
|
|
m_pLogger->INFO(buff.str());
|
|
|
}
|
|
|
if (!isvalid(imginfo) || (imginfo->channel!=1 && imginfo->channel!=3)) {
|
|
@@ -474,6 +474,12 @@ int CTeaSort::load_data(
|
|
|
else {
|
|
|
m_raw_img = imginfo2mat(imginfo);
|
|
|
}
|
|
|
+ if (m_pLogger) {
|
|
|
+ stringstream buff;
|
|
|
+ buff << "load image stream: " << m_imgId << m_dtype_str << "image, width=" << m_raw_img.cols
|
|
|
+ << "\theight=" << m_raw_img.rows << "\tchannels=" << m_raw_img.channels();
|
|
|
+ m_pLogger->INFO(buff.str());
|
|
|
+ }
|
|
|
|
|
|
}
|
|
|
else {
|
|
@@ -487,8 +493,8 @@ int CTeaSort::load_data(
|
|
|
}
|
|
|
if (m_pLogger) {
|
|
|
stringstream buff;
|
|
|
- buff << m_imgId << m_dtype_str << "image, width=" << img.cols
|
|
|
- << "\theight=" << img.rows;
|
|
|
+ buff <<"read image file: "<< m_imgId << m_dtype_str << "image, width=" << img.cols
|
|
|
+ << "\theight=" << img.rows << "\tchannels=" << img.channels();
|
|
|
m_pLogger->INFO(buff.str());
|
|
|
}
|
|
|
|
|
@@ -497,6 +503,12 @@ int CTeaSort::load_data(
|
|
|
//image saver
|
|
|
if (m_ppImgSaver && *m_ppImgSaver) {
|
|
|
(*m_ppImgSaver)->saveImage(m_raw_img, m_imgId);
|
|
|
+ if (m_pLogger) {
|
|
|
+ stringstream buff;
|
|
|
+ buff <<"saved: "<< m_imgId << m_dtype_str << "image, width=" << m_raw_img.cols
|
|
|
+ << "\theight=" << m_raw_img.rows<<"\tchannels="<< m_raw_img.channels();
|
|
|
+ m_pLogger->INFO(buff.str());
|
|
|
+ }
|
|
|
}
|
|
|
return rst;
|
|
|
}
|