tami/MYNT-EYE-S-SDK
6
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(sample): record add more stream enum support.

Browse Source
This commit is contained in:
TinyO 2019-10-22 12:39:20 +08:00
parent 6c70c0bf74
commit 85c8853540
1 changed files with 34 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
samples/dataset.cc
View File

@ -166,6 +166,15 @@ Dataset::writer_t Dataset::GetStreamWriter(const Stream &stream) {
case Stream::DISPARITY: { case Stream::DISPARITY: {
writer->outdir = outdir_ + MYNTEYE_OS_SEP "disparity"; writer->outdir = outdir_ + MYNTEYE_OS_SEP "disparity";
} break; } break;
case Stream::RIGHT_RECTIFIED: {
writer->outdir = outdir_ + MYNTEYE_OS_SEP "right_rect";
} break;
case Stream::LEFT_RECTIFIED: {
writer->outdir = outdir_ + MYNTEYE_OS_SEP "left_rect";
} break;
case Stream::DISPARITY_NORMALIZED: {
writer->outdir = outdir_ + MYNTEYE_OS_SEP "disparity_norm";
} break;
default: default:
LOG(FATAL) << "Unsupported stream: " << stream; LOG(FATAL) << "Unsupported stream: " << stream;
} }
@ -176,7 +185,31 @@ Dataset::writer_t Dataset::GetStreamWriter(const Stream &stream) {
writer->ofs << "seq, frame_id, timestamp, exposure_time" << std::endl; writer->ofs << "seq, frame_id, timestamp, exposure_time" << std::endl;
writer->ofs << FULL_PRECISION; writer->ofs << FULL_PRECISION;
stream_writers_[stream] = writer; stream_writers_[stream] = writer
return -1;
}
cv::imshow("frame", img);
if (img_count > 10 && imu_count > 50) { // save
// save Stream::LEFT
for (auto &&left : left_datas) {
dataset.SaveStreamData(Stream::LEFT, left);
}
// save Stream::RIGHT
// for (auto &&right : right_datas) {
// dataset.SaveStreamData(Stream::RIGHT, right);
// }
// save Stream::DEPTH
for (auto &&depth : depth_datas) {
dataset.SaveStreamData(Stream::DEPTH, depth);
}
// save Stream::DISPARITY
// for (auto &&disparity : disparity_datas) {
// dataset.SaveStreamData(Stream::DISPARITY, disparity);
// }
for (auto &&motion : motion_datas) {
dataset.SaveMotionData(motion);
};
stream_counts_[stream] = 0; stream_counts_[stream] = 0;
return writer; return writer;
} }