| 1144 | |
| 1145 | template<typename Dtype> |
| 1146 | void DataTransformer<Dtype>::Transform_nv(const Datum& datum, Blob<Dtype>* transformed_data, Blob<Dtype>* transformed_label, int cnt) { |
| 1147 | //std::cout << "Function 2 is used"; std::cout.flush(); |
| 1148 | // int offset = datum.height()*datum.width(); |
| 1149 | // int offset3 = 3 * offset; |
| 1150 | // int offset1 = datum.width(); |
| 1151 | // MetaData meta; |
| 1152 | // ReadMetaData(meta, datum.data(), offset3, offset1); |
| 1153 | // LOG(INFO) << "dataset: " << meta.dataset <<"; img_size: " << meta.img_size |
| 1154 | // << "; meta.annolist_index: " << meta.annolist_index; |
| 1155 | |
| 1156 | const int datum_channels = datum.channels(); |
| 1157 | //LOG(INFO) << datum.channels(); |
| 1158 | //const int datum_height = datum.height(); |
| 1159 | //const int datum_width = datum.width(); |
| 1160 | |
| 1161 | const int im_channels = transformed_data->channels(); |
| 1162 | //LOG(INFO) << im_channels; |
| 1163 | //const int im_height = transformed_data->height(); |
| 1164 | //const int im_width = transformed_data->width(); |
| 1165 | const int im_num = transformed_data->num(); |
| 1166 | |
| 1167 | //const int lb_channels = transformed_label->channels(); |
| 1168 | //const int lb_height = transformed_label->height(); |
| 1169 | //const int lb_width = transformed_label->width(); |
| 1170 | const int lb_num = transformed_label->num(); |
| 1171 | |
| 1172 | //LOG(INFO) << "image shape: " << transformed_data->num() << " " << transformed_data->channels() << " " |
| 1173 | // << transformed_data->height() << " " << transformed_data->width(); |
| 1174 | //LOG(INFO) << "label shape: " << transformed_label->num() << " " << transformed_label->channels() << " " |
| 1175 | // << transformed_label->height() << " " << transformed_label->width(); |
| 1176 | |
| 1177 | CHECK_EQ(datum_channels, 6); |
| 1178 | CHECK_EQ(im_channels, 6); |
| 1179 | //CHECK_EQ(im_channels, 4); |
| 1180 | //CHECK_EQ(datum_channels, 4); |
| 1181 | //CHECK_EQ(im_channels, 5); |
| 1182 | //CHECK_EQ(datum_channels, 5); |
| 1183 | |
| 1184 | CHECK_EQ(im_num, lb_num); |
| 1185 | //CHECK_LE(im_height, datum_height); |
| 1186 | //CHECK_LE(im_width, datum_width); |
| 1187 | CHECK_GE(im_num, 1); |
| 1188 | |
| 1189 | //const int crop_size = param_.crop_size(); |
| 1190 | |
| 1191 | // if (crop_size) { |
| 1192 | // CHECK_EQ(crop_size, im_height); |
| 1193 | // CHECK_EQ(crop_size, im_width); |
| 1194 | // } else { |
| 1195 | // CHECK_EQ(datum_height, im_height); |
| 1196 | // CHECK_EQ(datum_width, im_width); |
| 1197 | // } |
| 1198 | |
| 1199 | Dtype* transformed_data_pointer = transformed_data->mutable_cpu_data(); |
| 1200 | Dtype* transformed_label_pointer = transformed_label->mutable_cpu_data(); |
| 1201 | |
| 1202 | Transform_nv(datum, transformed_data_pointer, transformed_label_pointer, cnt); //call function 1 |
| 1203 | } |