read and save
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 | #include "util/image_util.h" #ifdef USE_PANGOLIN_VIEWER #include "pangolin_viewer/viewer.h" #elif USE_SOCKET_PUBLISHER #include "socket_publisher/publisher.h" #endif #include "openvslam/system.h" #include "openvslam/config.h" #include <iostream> #include <chrono> #include <numeric> #include <opencv2/core/core.hpp> #include <opencv2/imgcodecs.hpp> #include <spdlog/spdlog.h> #include <popl.hpp> / / 打印精度控制 #include <iostream> #include <iomanip> using namespace std; #ifdef USE_STACK_TRACE_LOGGER #include <glog/logging.h> #endif #ifdef USE_GOOGLE_PERFTOOLS #include <gperftools/profiler.h> #endif void LoadImages(const string &strPathToSequence, vector<string> &vstrImageFilenames, vector<string> &vTimestamps, vector<vector<double>> &GPSDataList) { ifstream fTimes; string strPathTimeFile = strPathToSequence + "/FHY_config/FHY_gps.txt" ; / / load image list FILE * file ; file = std::fopen(strPathTimeFile.c_str() , "r" ); / / .c_str() if ( file = = NULL){ printf( "cannot find file: %s \n" , strPathTimeFile.c_str()); return ; } char timeshara_c[ 30 ]; char lat_c[ 30 ], lon_c[ 30 ], alt_c[ 30 ]; vector<string> imageNameTimeList; / / vector<vector<string>> GPSDataList; / / 按照string 读取出来 容易保存超高精度 while (fscanf( file , "%s %s %s %s" , ×hara_c, &lat_c, &lon_c, &alt_c) ! = EOF) / / / / % lf之间应该有逗号,因为没有逗号只能读第一个数。用&是因为要把数存到对应数组元素的地址中去。\n是换行读取 / / while (fscanf( file , "%lf %lf %lf %lf" , ×harap_l, &lat_l, &lon_l, &alt_l) ! = EOF) { string timesharap_s; string lat_s, lon_s, alt_s; timesharap_s = (string(timeshara_c)); / / stod lat_s = (string(lat_c)); lon_s = (string(lon_c)); alt_s = (string(alt_c)); double timesharap_d; double lat_d, lon_d, alt_d; timesharap_d = std::stod(timesharap_s); lat_d = std::stod(lat_s); lon_d = std::stod(lon_s); alt_d = std::stod(alt_s); imageNameTimeList.push_back(timesharap_s); vTimestamps.push_back(timesharap_s); / / cout<< "timesharap_s " <<timesharap_s<< " timesharap_d " <<timesharap_d<<endl; vector<double> gpsdata_i; gpsdata_i.push_back(timesharap_d); gpsdata_i.push_back(lat_d); gpsdata_i.push_back(lon_d); gpsdata_i.push_back(alt_d); GPSDataList.push_back(gpsdata_i); std::cout<< fixed << setprecision( 18 ) << "timesharap_string " <<timesharap_s<< " timesharap_double18 " <<timesharap_d << ' ' << setprecision( 18 ) <<lat_s<< " " << lon_s << " " <<alt_s<<std::endl; } std::fclose( file ); string strPrefixLeft = strPathToSequence + "/left/" ; const int nTimes = vTimestamps.size(); vstrImageFilenames.resize(nTimes); for ( int i = 0 ; i<nTimes; i + + ) { / / stringstream ss; / / ss << setfill( '0' ) << setw( 10 ) << i; vstrImageFilenames[i] = strPrefixLeft + imageNameTimeList[i] + ".png" ; / / cout<< vstrImageFilenames[i] <<endl; } } vector<string> vstrImageFilenames; / / 图像路径 vector<double> vTimestamps; / / 时间戳 也就是图像名字 vector<vector<double>> GPSDataList; / / 保存GPS数据列表 / / = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = 跟踪进程 = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = void mono_tracking(const std::shared_ptr<openvslam::config>& cfg, const std::string& vocab_file_path, const std::string& image_dir_path, const std::string& mask_img_path, const unsigned int frame_skip, const bool no_sleep, const bool auto_term, const bool eval_log, const std::string& map_db_path, std::string &data_path, vector<string> &vstrImageFilenames, vector<string> &vTimestamps, vector<vector<double>> &GPSDataList ) { / / load the mask image const cv::Mat mask = mask_img_path.empty() ? cv::Mat{} : cv::imread(mask_img_path, cv::IMREAD_GRAYSCALE); / / const image_sequence sequence(image_dir_path, cfg - >camera_ - >fps_); / / const auto frames = sequence.get_frames(); int nImages = vstrImageFilenames.size(); / / build a SLAM system openvslam::system SLAM(cfg, vocab_file_path); / / startup the SLAM process SLAM.startup(); #ifdef USE_PANGOLIN_VIEWER pangolin_viewer::viewer viewer(cfg, &SLAM, SLAM.get_frame_publisher(), SLAM.get_map_publisher()); #elif USE_SOCKET_PUBLISHER socket_publisher::publisher publisher(cfg, &SLAM, SLAM.get_frame_publisher(), SLAM.get_map_publisher()); #endif std::vector<double> track_times; track_times.reserve(nImages); / / run the SLAM in another thread std::thread thread([&]() { string path = data_path + "/openvslam_outdata/slam_xyzgps.txt" ; std::ofstream ofs(path, std::ios::out); if (!ofs.is_open()) { spdlog::critical( "cannot create a file at {}" , path); throw std::runtime_error( "cannot create a file at " + path); } for (unsigned int i = 0 ; i < nImages; + + i) { / / const auto& frame = frames.at(i); double timesharpi_fps = ( 1.0 / cfg - >camera_ - >fps_) * i; / / 根据帧率控制读取离线图像的速度 double timesharpi1_fps = ( 1.0 / cfg - >camera_ - >fps_) * (i + 1 ); const auto img = cv::imread(vstrImageFilenames[i], cv::IMREAD_UNCHANGED); const auto tp_1 = std::chrono::steady_clock::now(); if (!img.empty() && (i % frame_skip = = 0 )) { / / input the current frame and estimate the camera pose / / = = = = = = = = = = = = = = = = = = = = = = = 识别当前帧 = = = = = = = = = = = = = = = = / / 这里的位姿并没有经历过局部和全局优化 / / double img_time_sharp = std::stod(vTimestamps[i]); / / time will lost no img name double img_time_sharp = i; Eigen::Matrix4d cam_pose_cw = SLAM.feed_monocular_frame(img, img_time_sharp , mask); Eigen::Matrix4d cam_pose_wc = cam_pose_cw.inverse(); Eigen::Matrix3d rot_wc = cam_pose_wc.block< 3 , 3 >( 0 , 0 ); Eigen::Vector3d trans_wc = cam_pose_wc.block< 3 , 1 >( 0 , 3 ); Eigen::Quaterniond quat_wc = Eigen::Quaterniond(rot_wc); cout << fixed << std::setprecision( 4 ) / / << vTimestamps[i] << " " << i << " " << std::setprecision( 9 ) << trans_wc( 0 ) << " " << trans_wc( 1 ) << " " << trans_wc( 2 ) << " " / / << quat_wc.x() << " " << quat_wc.y() << " " << quat_wc.z() << " " << quat_wc.w() << std::endl; << GPSDataList[i][ 1 ] << " " << GPSDataList[i][ 2 ] << " " << GPSDataList[i][ 3 ] << std::endl; ofs << fixed << std::setprecision( 4 ) / / << vTimestamps[i] << " " << i << " " << std::setprecision( 9 ) << trans_wc( 0 ) << " " << trans_wc( 1 ) << " " << trans_wc( 2 ) << " " << GPSDataList[i][ 1 ] << " " << GPSDataList[i][ 2 ] << " " << GPSDataList[i][ 3 ] << std::endl; } const auto tp_2 = std::chrono::steady_clock::now(); const auto track_time = std::chrono::duration_cast<std::chrono::duration<double>>(tp_2 - tp_1).count(); if (i % frame_skip = = 0 ) { track_times.push_back(track_time); } / / wait until the timestamp of the next frame if (!no_sleep && i < nImages - 1 ) { / / / / 根据帧率控制读取离线图像的速度 const auto wait_time = timesharpi1_fps - (timesharpi_fps + track_time); if ( 0.0 < wait_time) { / / std::this_thread::sleep_for(std::chrono::microseconds(static_cast<unsigned int >(wait_time * 1e6 ))); } } / / check if the termination of SLAM system is requested or not if (SLAM.terminate_is_requested()) { break ; } } / / wait until the loop BA is finished while (SLAM.loop_BA_is_running()) { std::this_thread::sleep_for(std::chrono::microseconds( 5000 )); } / / automatically close the viewer #ifdef USE_PANGOLIN_VIEWER if (auto_term) { viewer.request_terminate(); } #elif USE_SOCKET_PUBLISHER if (auto_term) { publisher.request_terminate(); } #endif ofs.close(); }); / / run the viewer in the current thread #ifdef USE_PANGOLIN_VIEWER viewer.run(); #elif USE_SOCKET_PUBLISHER publisher.run(); #endif thread.join(); / / shutdown the SLAM process SLAM.shutdown(); if ( 1 ) { / / output the trajectories for evaluation SLAM.save_frame_trajectory(data_path + "/openvslam_outdata/frame_trajectory.txt" , "TUM" ); SLAM.save_keyframe_trajectory(data_path + "/openvslam_outdata/keyframe_trajectory.txt" , "TUM" ); / / output the tracking times for evaluation std::ofstream ofs(data_path + "/openvslam_outdata/track_times.txt" , std::ios::out); if (ofs.is_open()) { for (const auto track_time : track_times) { ofs << track_time << std::endl; } ofs.close(); } } if (!map_db_path.empty()) { / / output the map database SLAM.save_map_database(map_db_path); } std::sort(track_times.begin(), track_times.end()); const auto total_track_time = std::accumulate(track_times.begin(), track_times.end(), 0.0 ); std::cout << "median tracking time: " << track_times.at(track_times.size() / 2 ) << "[s]" << std::endl; std::cout << "mean tracking time: " << total_track_time / track_times.size() << "[s]" << std::endl; } / / = = = = = = = = = = = = = = = = = = = = = = = = 主函数 = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = int main( int argc, char * argv[]) { #ifdef USE_STACK_TRACE_LOGGER google::InitGoogleLogging(argv[ 0 ]); google::InstallFailureSignalHandler(); #endif / / create options / / # 参数输入有顺序 别搞错位了 / / popl::OptionParser op( "Allowed options" ); / / auto help = op.add<popl::Switch>( "h" , "help" , "produce help message" ); / / auto vocab_file_path = op.add<popl::Value<std::string>>( "v" , "vocab" , "vocabulary file path" ); / / auto img_dir_path = op.add<popl::Value<std::string>>( "i" , "img-dir" , "directory path which contains images" ); / / auto config_file_path = op.add<popl::Value<std::string>>( "c" , "config" , "config file path" ); / / auto mask_img_path = op.add<popl::Value<std::string>>(" ", " mask ", " mask image path ", " "); / / auto frame_skip = op.add<popl::Value<unsigned int >>(" ", " frame - skip ", " interval of frame skip", 1 ); / / auto no_sleep = op.add<popl::Switch>(" ", " no - sleep ", " not wait for next frame in real time"); / / auto auto_term = op.add<popl::Switch>(" ", " auto - term ", " automatically terminate the viewer"); / / auto debug_mode = op.add<popl::Switch>(" ", " debug ", " debug mode"); / / auto eval_log = op.add<popl::Switch>(" ", " eval - log ", " store trajectory and tracking times for evaluation"); / / auto map_db_path = op.add<popl::Value<std::string>>( "p" , "map-db" , "store a map database at this path after SLAM" , ""); string vocab_file_path = argv[ 1 ]; string data_dir_path = argv[ 2 ]; std::string img_dir_path = data_dir_path + "/left/" ; std::string map_db_path = data_dir_path + "/openvslam_outdata/" + argv[ 4 ]; string config_file_path = argv[ 3 ]; string mask_img_path = ""; unsigned int frame_skip = 1 ; bool no_sleep = 0 ; / / 睡眠等待处理完在加载下一张离线图像 bool auto_term = 0 ; bool debug_mode = 1 ; bool eval_log = 1 ; if (vocab_file_path = = " " || img_dir_path==" " || config_file_path==" ") { std::cerr << "invalid arguments" << std::endl; std::cerr << std::endl; / / std::cerr << op << std::endl; return EXIT_FAILURE; } / / setup logger spdlog::set_pattern( "[%Y-%m-%d %H:%M:%S.%e] %^[%L] %v%$" ); if (debug_mode) { spdlog::set_level(spdlog::level::debug); } else { spdlog::set_level(spdlog::level::info); } / / load configuration std::shared_ptr<openvslam::config> cfg; try { cfg = std::make_shared<openvslam::config>(config_file_path); } catch (const std::exception& e) { std::cerr << e.what() << std::endl; return EXIT_FAILURE; } #ifdef USE_GOOGLE_PERFTOOLS ProfilerStart( "slam.prof" ); #endif / / = = = = = = = = = = = = = = = = = 加载图像数据 gps时间戳 = = = = = = = = = = = = = = = = = = = = = = = = = = / / Retrieve paths to images vector<string> vstrImageFilenames; / / 图像路径 vector<string> vTimestamps_string; / / 时间戳 也就是图像名字 vector<vector<double>> GPSDataList; / / 保存GPS数据列表 / / 根据时间戳列表 找图像加载 LoadImages(data_dir_path, vstrImageFilenames, vTimestamps_string, GPSDataList); / / run tracking if (cfg - >camera_ - >setup_type_ = = openvslam::camera::setup_type_t::Monocular) { mono_tracking(cfg, vocab_file_path, img_dir_path, mask_img_path, frame_skip, no_sleep, auto_term, eval_log, map_db_path, data_dir_path, vstrImageFilenames, vTimestamps_string, GPSDataList ); } else { throw std::runtime_error( "Invalid setup type: " + cfg - >camera_ - >get_setup_type_string()); } #ifdef USE_GOOGLE_PERFTOOLS ProfilerStop(); #endif return EXIT_SUCCESS; } |
save
yi ding yao
1 | ofs << fixed |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | string path = image_path + "/ORB_SLAM3_out/slam_xyzgps.txt" ; std::ofstream ofs(path, std::ios::out); if (!ofs.is_open()) { cout << " error path" << path << endl; } ofs << fixed << std::setprecision( 4 ) / / << timestamps.at(frm_id) << " " << ni << " " << std::setprecision( 8 ) << t_wc( 0 ) << " " << t_wc( 1 ) << " " << t_wc( 2 ) << " " << GPSDataList[ni][ 1 ] << " " << GPSDataList[ni][ 2 ] << " " << GPSDataList[ni][ 3 ] << " " << q.x() << " " << q.y() << " " << q.z() << " " << q.w() << endl; ofs.close(); |
分类:
C++教程 (2) 实用样例
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 开源Multi-agent AI智能体框架aevatar.ai,欢迎大家贡献代码
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY