/** * This file is part of ORB-SLAM3 * * Copyright (C) 2017-2020 Carlos Campos, Richard Elvira, Juan J. Gómez Rodríguez, José M.M. Montiel and Juan D. Tardós, University of Zaragoza. * Copyright (C) 2014-2016 Raúl Mur-Artal, José M.M. Montiel and Juan D. Tardós, University of Zaragoza. * * ORB-SLAM3 is free software: you can redistribute it and/or modify it under the terms of the GNU General Public * License as published by the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * ORB-SLAM3 is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even * the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License along with ORB-SLAM3. * If not, see . */ #include "KeyFrame.h" #include "Converter.h" #include "ORBmatcher.h" #include "ImuTypes.h" #include namespace ORB_SLAM3 { long unsigned int KeyFrame::nNextId=0; KeyFrame::KeyFrame(): mnFrameId(0), mTimeStamp(0), mnGridCols(FRAME_GRID_COLS), mnGridRows(FRAME_GRID_ROWS), mfGridElementWidthInv(0), mfGridElementHeightInv(0), mnTrackReferenceForFrame(0), mnFuseTargetForKF(0), mnBALocalForKF(0), mnBAFixedForKF(0), mnBALocalForMerge(0), mnLoopQuery(0), mnLoopWords(0), mnRelocQuery(0), mnRelocWords(0), mnMergeQuery(0), mnMergeWords(0), mnBAGlobalForKF(0), fx(0), fy(0), cx(0), cy(0), invfx(0), invfy(0), mnPlaceRecognitionQuery(0), mnPlaceRecognitionWords(0), mPlaceRecognitionScore(0), mbf(0), mb(0), mThDepth(0), N(0), mvKeys(static_cast >(NULL)), mvKeysUn(static_cast >(NULL)), mvuRight(static_cast >(NULL)), mvDepth(static_cast >(NULL)), /*mDescriptors(NULL),*/ /*mBowVec(NULL), mFeatVec(NULL),*/ mnScaleLevels(0), mfScaleFactor(0), mfLogScaleFactor(0), mvScaleFactors(0), mvLevelSigma2(0), mvInvLevelSigma2(0), mnMinX(0), mnMinY(0), mnMaxX(0), mnMaxY(0), /*mK(NULL),*/ mPrevKF(static_cast(NULL)), mNextKF(static_cast(NULL)), mbFirstConnection(true), mpParent(NULL), mbNotErase(false), mbToBeErased(false), mbBad(false), mHalfBaseline(0), mbCurrentPlaceRecognition(false), mbHasHessian(false), mnMergeCorrectedForKF(0), NLeft(0),NRight(0), mnNumberOfOpt(0) { } KeyFrame::KeyFrame(Frame &F, Map *pMap, KeyFrameDatabase *pKFDB): bImu(pMap->isImuInitialized()), mnFrameId(F.mnId), mTimeStamp(F.mTimeStamp), mnGridCols(FRAME_GRID_COLS), mnGridRows(FRAME_GRID_ROWS), mfGridElementWidthInv(F.mfGridElementWidthInv), mfGridElementHeightInv(F.mfGridElementHeightInv), mnTrackReferenceForFrame(0), mnFuseTargetForKF(0), mnBALocalForKF(0), mnBAFixedForKF(0), mnBALocalForMerge(0), mnLoopQuery(0), mnLoopWords(0), mnRelocQuery(0), mnRelocWords(0), mnBAGlobalForKF(0), mnPlaceRecognitionQuery(0), mnPlaceRecognitionWords(0), mPlaceRecognitionScore(0), fx(F.fx), fy(F.fy), cx(F.cx), cy(F.cy), invfx(F.invfx), invfy(F.invfy), mbf(F.mbf), mb(F.mb), mThDepth(F.mThDepth), N(F.N), mvKeys(F.mvKeys), mvKeysUn(F.mvKeysUn), mvuRight(F.mvuRight), mvDepth(F.mvDepth), mDescriptors(F.mDescriptors.clone()), mBowVec(F.mBowVec), mFeatVec(F.mFeatVec), mnScaleLevels(F.mnScaleLevels), mfScaleFactor(F.mfScaleFactor), mfLogScaleFactor(F.mfLogScaleFactor), mvScaleFactors(F.mvScaleFactors), mvLevelSigma2(F.mvLevelSigma2), mvInvLevelSigma2(F.mvInvLevelSigma2), mnMinX(F.mnMinX), mnMinY(F.mnMinY), mnMaxX(F.mnMaxX), mnMaxY(F.mnMaxY), mK(F.mK), mPrevKF(NULL), mNextKF(NULL), mpImuPreintegrated(F.mpImuPreintegrated), mImuCalib(F.mImuCalib), mvpMapPoints(F.mvpMapPoints), mpKeyFrameDB(pKFDB), mpORBvocabulary(F.mpORBvocabulary), mbFirstConnection(true), mpParent(NULL), mDistCoef(F.mDistCoef), mbNotErase(false), mnDataset(F.mnDataset), mbToBeErased(false), mbBad(false), mHalfBaseline(F.mb/2), mpMap(pMap), mbCurrentPlaceRecognition(false), mNameFile(F.mNameFile), mbHasHessian(false), mnMergeCorrectedForKF(0), mpCamera(F.mpCamera), mpCamera2(F.mpCamera2), mvLeftToRightMatch(F.mvLeftToRightMatch),mvRightToLeftMatch(F.mvRightToLeftMatch),mTlr(F.mTlr.clone()), mvKeysRight(F.mvKeysRight), NLeft(F.Nleft), NRight(F.Nright), mTrl(F.mTrl), mnNumberOfOpt(0) { imgLeft = F.imgLeft.clone(); imgRight = F.imgRight.clone(); mnId=nNextId++; // 根据指定的普通帧, 初始化用于加速匹配的网格对象信息; 其实就把每个网格中有的特征点的索引复制过来 mGrid.resize(mnGridCols); if(F.Nleft != -1) mGridRight.resize(mnGridCols); for(int i=0; iGetId(); } void KeyFrame::ComputeBoW() { // 只有当词袋向量或者节点和特征序号的特征向量为空的时候执行 if(mBowVec.empty() || mFeatVec.empty()) { vector vCurrentDesc = Converter::toDescriptorVector(mDescriptors); // Feature vector associate features with nodes in the 4th level (from leaves up) // We assume the vocabulary tree has 6 levels, change the 4 otherwise mpORBvocabulary->transform(vCurrentDesc,mBowVec,mFeatVec,4); } } // 设置当前关键帧的位姿 void KeyFrame::SetPose(const cv::Mat &Tcw_) { unique_lock lock(mMutexPose); Tcw_.copyTo(Tcw); cv::Mat Rcw = Tcw.rowRange(0,3).colRange(0,3); cv::Mat tcw = Tcw.rowRange(0,3).col(3); cv::Mat Rwc = Rcw.t(); Ow = -Rwc*tcw; if (!mImuCalib.Tcb.empty()) Owb = Rwc*mImuCalib.Tcb.rowRange(0,3).col(3)+Ow; // 计算当前位姿的逆 Twc = cv::Mat::eye(4,4,Tcw.type()); Rwc.copyTo(Twc.rowRange(0,3).colRange(0,3)); Ow.copyTo(Twc.rowRange(0,3).col(3)); // center为相机坐标系(左目)下,立体相机中心的坐标 // 立体相机中心点坐标与左目相机坐标之间只是在x轴上相差mHalfBaseline, // 因此可以看出,立体相机中两个摄像头的连线为x轴,正方向为左目相机指向右目相机 (齐次坐标) cv::Mat center = (cv::Mat_(4,1) << mHalfBaseline, 0 , 0, 1); // 世界坐标系下,左目相机中心到立体相机中心的向量,方向由左目相机指向立体相机中心 Cw = Twc*center; } void KeyFrame::SetVelocity(const cv::Mat &Vw_) { unique_lock lock(mMutexPose); Vw_.copyTo(Vw); } // 获取位姿 cv::Mat KeyFrame::GetPose() { unique_lock lock(mMutexPose); return Tcw.clone(); } // 获取位姿的逆 cv::Mat KeyFrame::GetPoseInverse() { unique_lock lock(mMutexPose); return Twc.clone(); } // 获取(左目)相机的中心在世界坐标系下的坐标 cv::Mat KeyFrame::GetCameraCenter() { unique_lock lock(mMutexPose); return Ow.clone(); } // 获取双目相机的中心,这个只有在可视化的时候才会用到 cv::Mat KeyFrame::GetStereoCenter() { unique_lock lock(mMutexPose); return Cw.clone(); } cv::Mat KeyFrame::GetImuPosition() { unique_lock lock(mMutexPose); return Owb.clone(); } cv::Mat KeyFrame::GetImuRotation() { unique_lock lock(mMutexPose); return Twc.rowRange(0,3).colRange(0,3)*mImuCalib.Tcb.rowRange(0,3).colRange(0,3); } cv::Mat KeyFrame::GetImuPose() { unique_lock lock(mMutexPose); return Twc*mImuCalib.Tcb; } cv::Mat KeyFrame::GetRotation() { unique_lock lock(mMutexPose); return Tcw.rowRange(0,3).colRange(0,3).clone(); } // 获取位置 cv::Mat KeyFrame::GetTranslation() { unique_lock lock(mMutexPose); return Tcw.rowRange(0,3).col(3).clone(); } cv::Mat KeyFrame::GetVelocity() { unique_lock lock(mMutexPose); return Vw.clone(); } // 为关键帧之间添加或更新连接 void KeyFrame::AddConnection(KeyFrame *pKF, const int &weight) { { // 如果被占用就一直等着,这个添加连接的操作不能够被放弃 unique_lock lock(mMutexConnections); // 判断当前关键帧是否已经和其他的关键帧创建了联系 // std::map::count函数只可能返回0或1两种情况 if(!mConnectedKeyFrameWeights.count(pKF)) // count函数返回0,mConnectedKeyFrameWeights中没有pKF,之前没有连接 mConnectedKeyFrameWeights[pKF]=weight; else if(mConnectedKeyFrameWeights[pKF]!=weight) // 之前连接的权重不一样,更新 mConnectedKeyFrameWeights[pKF]=weight; else return; } // 如果添加了更新的连接关系就要更新一下,主要是重新进行排序 UpdateBestCovisibles(); } /** * @brief 按照权重对连接的关键帧进行排序 * * 更新后的变量存储在mvpOrderedConnectedKeyFrames和mvOrderedWeights中 */ void KeyFrame::UpdateBestCovisibles() { unique_lock lock(mMutexConnections); vector > vPairs; vPairs.reserve(mConnectedKeyFrameWeights.size()); // 取出所有连接的关键帧,mConnectedKeyFrameWeights的类型为std::map,而vPairs变量将共视的3D点数放在前面,利于排序 for(map::iterator mit=mConnectedKeyFrameWeights.begin(), mend=mConnectedKeyFrameWeights.end(); mit!=mend; mit++) vPairs.push_back(make_pair(mit->second,mit->first)); // 按照权重进行排序(默认是从小到大) sort(vPairs.begin(),vPairs.end()); // 为什么要用链表保存?因为插入和删除操作方便,只需要修改上一节点位置,不需要移动其他元素 list lKFs; // keyframe list lWs; // weight for(size_t i=0, iend=vPairs.size(); i(lKFs.begin(),lKFs.end()); mvOrderedWeights = vector(lWs.begin(), lWs.end()); } // 得到与该关键帧连接(>15个共视地图点)的关键帧(没有排序的) set KeyFrame::GetConnectedKeyFrames() { unique_lock lock(mMutexConnections); set s; for(map::iterator mit=mConnectedKeyFrameWeights.begin();mit!=mConnectedKeyFrameWeights.end();mit++) s.insert(mit->first); return s; } // 得到与该关键帧连接的关键帧(已按权值排序) vector KeyFrame::GetVectorCovisibleKeyFrames() { unique_lock lock(mMutexConnections); return mvpOrderedConnectedKeyFrames; } // 得到与该关键帧连接的前N个关键帧(已按权值排序) vector KeyFrame::GetBestCovisibilityKeyFrames(const int &N) { unique_lock lock(mMutexConnections); if((int)mvpOrderedConnectedKeyFrames.size()(mvpOrderedConnectedKeyFrames.begin(),mvpOrderedConnectedKeyFrames.begin()+N); } // 得到与该关键帧连接的权重大于等于w的关键帧 vector KeyFrame::GetCovisiblesByWeight(const int &w) { unique_lock lock(mMutexConnections); // 如果没有和当前关键帧连接的关键帧 if(mvpOrderedConnectedKeyFrames.empty()) { return vector(); } // 从mvOrderedWeights找出第一个大于w的那个迭代器 vector::iterator it = upper_bound( mvOrderedWeights.begin(), //起点 mvOrderedWeights.end(), //终点 w, //目标阈值 KeyFrame::weightComp); //比较函数,由于我们这里从大到小排序,所以需要重定义比较函数 // 如果没有找到(最大的权重也比给定的阈值小) if(it==mvOrderedWeights.end() && mvOrderedWeights.back() < w) { return vector(); } else { int n = it-mvOrderedWeights.begin(); //cout << "n = " << n << endl; return vector(mvpOrderedConnectedKeyFrames.begin(), mvpOrderedConnectedKeyFrames.begin()+n); } } // 得到该关键帧与pKF的权重 int KeyFrame::GetWeight(KeyFrame *pKF) { unique_lock lock(mMutexConnections); if(mConnectedKeyFrameWeights.count(pKF)) return mConnectedKeyFrameWeights[pKF]; else // 没有连接的话权重也就是共视点个数就是0 return 0; } int KeyFrame::GetNumberMPs() { unique_lock lock(mMutexFeatures); int numberMPs = 0; for(size_t i=0, iend=mvpMapPoints.size(); i lock(mMutexFeatures); mvpMapPoints[idx]=pMP; } /** * @brief 由于其他的原因,导致当前关键帧观测到的某个地图点被删除(bad==true)了,将该地图点置为NULL * * @param[in] idx 地图点在该关键帧中的id */ void KeyFrame::EraseMapPointMatch(const int &idx) { unique_lock lock(mMutexFeatures); mvpMapPoints[idx]=static_cast(NULL); } void KeyFrame::EraseMapPointMatch(MapPoint* pMP) { tuple indexes = pMP->GetIndexInKeyFrame(this); size_t leftIndex = get<0>(indexes), rightIndex = get<1>(indexes); if(leftIndex != -1) mvpMapPoints[leftIndex]=static_cast(NULL); if(rightIndex != -1) mvpMapPoints[rightIndex]=static_cast(NULL); } // 地图点的替换 void KeyFrame::ReplaceMapPointMatch(const int &idx, MapPoint* pMP) { mvpMapPoints[idx]=pMP; } // 获取当前关键帧中的所有地图点 set KeyFrame::GetMapPoints() { unique_lock lock(mMutexFeatures); set s; for(size_t i=0, iend=mvpMapPoints.size(); iisBad()) s.insert(pMP); } return s; } // 关键帧中,大于等于最少观测数目minObs的MapPoints的数量.这些特征点被认为追踪到了 int KeyFrame::TrackedMapPoints(const int &minObs) { unique_lock lock(mMutexFeatures); int nPoints=0; // 是否检查数目 const bool bCheckObs = minObs>0; // N是当前帧中特征点的个数 for(int i=0; iisBad()) //并且不是坏点 { if(bCheckObs) { // 满足输入阈值要求的地图点计数加1 if(mvpMapPoints[i]->Observations()>=minObs) nPoints++; } else nPoints++; } } } return nPoints; } // 获取当前关键帧的具体的地图点 vector KeyFrame::GetMapPointMatches() { unique_lock lock(mMutexFeatures); return mvpMapPoints; } // 获取当前关键帧的具体的某个地图点 MapPoint* KeyFrame::GetMapPoint(const size_t &idx) { unique_lock lock(mMutexFeatures); return mvpMapPoints[idx]; } /* * 更新图的连接 * * 1. 首先获得该关键帧的所有MapPoint点,统计观测到这些3d点的每个关键帧与其它所有关键帧之间的共视程度 * 对每一个找到的关键帧,建立一条边,边的权重是该关键帧与当前关键帧公共3d点的个数。 * 2. 并且该权重必须大于一个阈值,如果没有超过该阈值的权重,那么就只保留权重最大的边(与其它关键帧的共视程度比较高) * 3. 对这些连接按照权重从大到小进行排序,以方便将来的处理 * 更新完covisibility图之后,如果没有初始化过,则初始化为连接权重最大的边(与其它关键帧共视程度最高的那个关键帧),类似于最大生成树 */ void KeyFrame::UpdateConnections(bool upParent) { // 在没有执行这个函数前,关键帧只和MapPoints之间有连接关系,这个函数可以更新关键帧之间的连接关系 map KFcounter; vector vpMP; { // 获得该关键帧的所有3D点 unique_lock lockMPs(mMutexFeatures); vpMP = mvpMapPoints; } //For all map points in keyframe check in which other keyframes are they seen //Increase counter for those keyframes // 通过3D点间接统计可以观测到这些3D点的所有关键帧之间的共视程度 // Step 1 统计每一个地图点都有多少关键帧与当前关键帧存在共视关系,统计结果放在KFcounter for(vector::iterator vit=vpMP.begin(), vend=vpMP.end(); vit!=vend; vit++) { MapPoint* pMP = *vit; if(!pMP) continue; if(pMP->isBad()) continue; // 对于每一个MapPoint点,observations记录了可以观测到该MapPoint的所有关键帧 map> observations = pMP->GetObservations(); for(map>::iterator mit=observations.begin(), mend=observations.end(); mit!=mend; mit++) { if(mit->first->mnId==mnId || mit->first->isBad() || mit->first->GetMap() != mpMap) continue; // 这里的操作非常精彩! // map[key] = value,当要插入的键存在时,会覆盖键对应的原来的值。如果键不存在,则添加一组键值对 // mit->first 是地图点看到的关键帧,同一个关键帧看到的地图点会累加到该关键帧计数 // 所以最后KFcounter 第一个参数表示某个关键帧,第2个参数表示该关键帧看到了多少当前帧的地图点,也就是共视程度 KFcounter[mit->first]++; } } // This should not happen if(KFcounter.empty()) return; //If the counter is greater than threshold add connection //In case no keyframe counter is over threshold add the one with maximum counter int nmax=0; KeyFrame* pKFmax=NULL; // 至少有15个共视地图点 int th = 15; // vPairs记录与其它关键帧共视帧数大于th的关键帧 // pair将关键帧的权重写在前面,关键帧写在后面方便后面排序 vector > vPairs; vPairs.reserve(KFcounter.size()); if(!upParent) cout << "UPDATE_CONN: current KF " << mnId << endl; // Step 2 找到对应权重最大的关键帧(共视程度最高的关键帧) for(map::iterator mit=KFcounter.begin(), mend=KFcounter.end(); mit!=mend; mit++) { if(!upParent) cout << " UPDATE_CONN: KF " << mit->first->mnId << " ; num matches: " << mit->second << endl; if(mit->second>nmax) { nmax=mit->second; pKFmax=mit->first; } if(mit->second>=th) { // 对应权重需要大于阈值,对这些关键帧建立连接 vPairs.push_back(make_pair(mit->second,mit->first)); // 对方关键帧也要添加这个信息 // 更新KFcounter中该关键帧的mConnectedKeyFrameWeights // 更新其它KeyFrame的mConnectedKeyFrameWeights,更新其它关键帧与当前帧的连接权重 (mit->first)->AddConnection(this,mit->second); } } // Step 3 如果没有连接到关键(超过阈值的权重),则对权重最大的关键帧建立连接 if(vPairs.empty()) { // 如果每个关键帧与它共视的关键帧的个数都少于th, // 那就只更新与其它关键帧共视程度最高的关键帧的mConnectedKeyFrameWeights // 这是对之前th这个阈值可能过高的一个补丁 vPairs.push_back(make_pair(nmax,pKFmax)); pKFmax->AddConnection(this,nmax); } // Step 4 对共视程度比较高的关键帧对更新连接关系及权重(从大到小) // vPairs里存的都是相互共视程度比较高的关键帧和共视权重,接下来由大到小进行排序 sort(vPairs.begin(),vPairs.end()); // sort函数默认升序排列 // 将排序后的结果分别组织成为两种数据类型 list lKFs; list lWs; for(size_t i=0; i lockCon(mMutexConnections); // mspConnectedKeyFrames = spConnectedKeyFrames; // 更新当前帧与其它关键帧的连接权重 mConnectedKeyFrameWeights = KFcounter; mvpOrderedConnectedKeyFrames = vector(lKFs.begin(),lKFs.end()); mvOrderedWeights = vector(lWs.begin(), lWs.end()); // if(mbFirstConnection && mnId!=mpMap->GetInitKFid()) // { // mpParent = mvpOrderedConnectedKeyFrames.front(); // mpParent->AddChild(this); // mbFirstConnection = false; // } // Step 5 更新生成树的连接 if(mbFirstConnection && mnId!=mpMap->GetInitKFid()) { /*if(!mpParent || mpParent->GetParent() != this) { KeyFrame* pBestParent = static_cast(NULL); for(KeyFrame* pKFi : mvpOrderedConnectedKeyFrames) { if(pKFi->GetParent() || pKFi->mnId == mpMap->GetInitKFid()) { pBestParent = pKFi; break; } } if(!pBestParent) { cout << "It can't be a covisible KF without Parent" << endl << endl; return; } mpParent = pBestParent; mpParent->AddChild(this); mbFirstConnection = false; }*/ // cout << "udt.conn.id: " << mnId << endl; // 初始化该关键帧的父关键帧为共视程度最高的那个关键帧 mpParent = mvpOrderedConnectedKeyFrames.front(); // 建立双向连接关系,将当前关键帧作为其子关键帧 mpParent->AddChild(this); mbFirstConnection = false; } } } // 添加子关键帧(即和子关键帧具有最大共视关系的关键帧就是当前关键帧) void KeyFrame::AddChild(KeyFrame *pKF) { unique_lock lockCon(mMutexConnections); mspChildrens.insert(pKF); } // 删除某个子关键帧 void KeyFrame::EraseChild(KeyFrame *pKF) { unique_lock lockCon(mMutexConnections); mspChildrens.erase(pKF); } // 改变当前关键帧的父关键帧 void KeyFrame::ChangeParent(KeyFrame *pKF) { unique_lock lockCon(mMutexConnections); // if(!mpParent && mpParent != this) // mpParent->EraseChild(this); if(pKF == this) { cout << "ERROR: Change parent KF, the parent and child are the same KF" << endl; throw std::invalid_argument("The parent and child can not be the same"); } mpParent = pKF; pKF->AddChild(this); } //获取当前关键帧的子关键帧 set KeyFrame::GetChilds() { unique_lock lockCon(mMutexConnections); return mspChildrens; } //获取当前关键帧的父关键帧 KeyFrame* KeyFrame::GetParent() { unique_lock lockCon(mMutexConnections); return mpParent; } // 判断某个关键帧是否是当前关键帧的子关键帧 bool KeyFrame::hasChild(KeyFrame *pKF) { unique_lock lockCon(mMutexConnections); return mspChildrens.count(pKF); } void KeyFrame::SetFirstConnection(bool bFirst) { unique_lock lockCon(mMutexConnections); mbFirstConnection=bFirst; } // 给当前关键帧添加回环边,回环边连接了形成闭环关系的关键帧 void KeyFrame::AddLoopEdge(KeyFrame *pKF) { unique_lock lockCon(mMutexConnections); mbNotErase = true; mspLoopEdges.insert(pKF); } // 获取和当前关键帧形成闭环关系的关键帧 set KeyFrame::GetLoopEdges() { unique_lock lockCon(mMutexConnections); return mspLoopEdges; } void KeyFrame::AddMergeEdge(KeyFrame* pKF) { unique_lock lockCon(mMutexConnections); mbNotErase = true; mspMergeEdges.insert(pKF); } set KeyFrame::GetMergeEdges() { unique_lock lockCon(mMutexConnections); return mspMergeEdges; } // 设置当前关键帧不要在优化的过程中被删除. 由回环检测线程调用 void KeyFrame::SetNotErase() { unique_lock lock(mMutexConnections); mbNotErase = true; } /** * @brief 删除当前的这个关键帧,表示不进行回环检测过程;由回环检测线程调用 * */ void KeyFrame::SetErase() { { unique_lock lock(mMutexConnections); // 如果当前关键帧和其他的关键帧没有形成回环关系,那么就删吧 if(mspLoopEdges.empty()) { mbNotErase = false; } } // mbToBeErased:删除之前记录的想要删但时机不合适没有删除的帧 if(mbToBeErased) { SetBadFlag(); } } /** * @brief 真正地执行删除关键帧的操作 * 需要删除的是该关键帧和其他所有帧、地图点之间的连接关系 * * mbNotErase作用:表示要删除该关键帧及其连接关系但是这个关键帧有可能正在回环检测或者计算sim3操作,这时候虽然这个关键帧冗余,但是却不能删除, * 仅设置mbNotErase为true,这时候调用setbadflag函数时,不会将这个关键帧删除,只会把mbTobeErase变成true,代表这个关键帧可以删除但不到时候,先记下来以后处理。 * 在闭环线程里调用 SetErase()会根据mbToBeErased 来删除之前可以删除还没删除的帧。 */ void KeyFrame::SetBadFlag() { // std::cout << "Erasing KF..." << std::endl; // Step 1 首先处理一下删除不了的特殊情况 { unique_lock lock(mMutexConnections); // 初始关键帧不能删除 if(mnId==mpMap->GetInitKFid()) { //std::cout << "KF.BADFLAG-> KF 0!!" << std::endl; return; } else if(mbNotErase) { // mbNotErase表示不应该删除,于是把mbToBeErased置为true,假装已经删除,其实没有删除 mbToBeErased = true; return; } if(!mpParent) { //cout << "KF.BADFLAG-> There is not parent, but it is not the first KF in the map" << endl; //cout << "KF.BADFLAG-> KF: " << mnId << "; first KF: " << mpMap->GetInitKFid() << endl; } } // Step 2 遍历所有和当前关键帧共视的关键帧,删除他们与当前关键帧的联系 for(map::iterator mit = mConnectedKeyFrameWeights.begin(), mend=mConnectedKeyFrameWeights.end(); mit!=mend; mit++) { mit->first->EraseConnection(this); // 让其它的KeyFrame删除与自己的联系 } //std::cout << "KF.BADFLAG-> Connection erased..." << std::endl; // Step 3 遍历每一个当前关键帧的地图点,删除每一个地图点和当前关键帧的联系 for(size_t i=0; iEraseObservation(this); // 让与自己有联系的MapPoint删除与自己的联系 // nDeletedPoints++; } } // cout << "nDeletedPoints: " << nDeletedPoints << endl; //std::cout << "KF.BADFLAG-> Observations deleted..." << std::endl; { unique_lock lock(mMutexConnections); unique_lock lock1(mMutexFeatures); // 清空自己与其它关键帧之间的联系 mConnectedKeyFrameWeights.clear(); mvpOrderedConnectedKeyFrames.clear(); // Update Spanning Tree // Step 4 更新生成树,主要是处理好父子关键帧,不然会造成整个关键帧维护的图断裂,或者混乱,不能够为后端提供较好的初值 // 子关键帧候选父关键帧 set sParentCandidates; // 将当前帧的父关键帧放入候选父关键帧 if(mpParent) sParentCandidates.insert(mpParent); //std::cout << "KF.BADFLAG-> Initially there are " << sParentCandidates.size() << " candidates" << std::endl; // Assign at each iteration one children with a parent (the pair with highest covisibility weight) // Include that children as new parent candidate for the rest // 如果这个关键帧有自己的子关键帧,告诉这些子关键帧,它们的父关键帧不行了,赶紧找新的父关键帧 while(!mspChildrens.empty()) { bool bContinue = false; int max = -1; KeyFrame* pC; KeyFrame* pP; // Step 4.1 遍历每一个子关键帧,让它们更新它们指向的父关键帧 for(set::iterator sit=mspChildrens.begin(), send=mspChildrens.end(); sit!=send; sit++) { KeyFrame* pKF = *sit; // 跳过无效的子关键帧 if(pKF->isBad()) continue; // Check if a parent candidate is connected to the keyframe // Step 4.2 子关键帧遍历每一个与它共视的关键帧 vector vpConnected = pKF->GetVectorCovisibleKeyFrames(); for(size_t i=0, iend=vpConnected.size(); i::iterator spcit=sParentCandidates.begin(), spcend=sParentCandidates.end(); spcit!=spcend; spcit++) { // Step 4.3 如果孩子和候选的父节点共视,说明找到了备选的父节点 if(vpConnected[i]->mnId == (*spcit)->mnId) { int w = pKF->GetWeight(vpConnected[i]); // 寻找并更新权值最大的那个共视关系 if(w>max) { pC = pKF; //子关键帧 pP = vpConnected[i]; //目前和子关键帧具有最大权值的关键帧(将来的父关键帧) max = w; //这个最大的权值 bContinue = true; //说明子节点找到了可以作为其新父关键帧的帧 } } } } } // Step 4.4 如果在上面的过程中找到了新的父节点 // 下面代码应该放到遍历子关键帧循环中? // 回答:不需要!这里while循环还没退出,会使用更新的sParentCandidates if(bContinue) { if(pC->mnId == pP->mnId) { /*cout << "ERROR: The parent and son can't be the same KF. ID: " << pC->mnId << endl; cout << "Current KF: " << mnId << endl; cout << "Parent of the map: " << endl;*/ } // 因为父节点死了,并且子节点找到了新的父节点,就把它更新为自己的父节点 pC->ChangeParent(pP); // 因为子节点找到了新的父节点并更新了父节点,那么该子节点升级,作为其它子节点的备选父节点 sParentCandidates.insert(pC); // 该子节点处理完毕,删掉 mspChildrens.erase(pC); } else break; } //std::cout << "KF.BADFLAG-> Apply change of parent to children" << std::endl; // If a children has no covisibility links with any parent candidate, assign to the original parent of this KF // Step 4.5 如果还有子节点没有找到新的父节点 if(!mspChildrens.empty()) { for(set::iterator sit=mspChildrens.begin(); sit!=mspChildrens.end(); sit++) { // 直接把父节点的父节点作为自己的父节点 即对于这些子节点来说,他们的新的父节点其实就是自己的爷爷节点 (*sit)->ChangeParent(mpParent); } } //std::cout << "KF.BADFLAG-> Apply change to its parent" << std::endl; if(mpParent){ mpParent->EraseChild(this); // 如果当前的关键帧要被删除的话就要计算这个,表示原父关键帧到当前关键帧的位姿变换 // 注意在这个删除的过程中,其实并没有将当前关键帧中存储的父关键帧的指针删除掉 mTcp = Tcw*mpParent->GetPoseInverse(); } else { //cout << "Error: KF haven't got a parent, it is imposible reach this code point without him" << endl; } // 标记当前关键帧已经死了 mbBad = true; } mpMap->EraseKeyFrame(this); mpKeyFrameDB->erase(this); } // 返回当前关键帧是否已经完蛋了 bool KeyFrame::isBad() { unique_lock lock(mMutexConnections); return mbBad; } // 删除当前关键帧和指定关键帧之间的共视关系 void KeyFrame::EraseConnection(KeyFrame* pKF) { // 其实这个应该表示是否真的是有共视关系 bool bUpdate = false; { unique_lock lock(mMutexConnections); if(mConnectedKeyFrameWeights.count(pKF)) { mConnectedKeyFrameWeights.erase(pKF); bUpdate=true; } } // 如果是真的有共视关系,那么删除之后就要更新共视关系 if(bUpdate) UpdateBestCovisibles(); } // 获取某个特征点的邻域中的特征点id,其实这个和 Frame.cc 中的那个函数基本上都是一致的; r为边长(半径) vector KeyFrame::GetFeaturesInArea(const float &x, const float &y, const float &r, const bool bRight) const { vector vIndices; vIndices.reserve(N); // 计算要搜索的cell的范围 float factorX = r; float factorY = r; // floor向下取整,mfGridElementWidthInv 为每个像素占多少个格子 const int nMinCellX = max(0,(int)floor((x-mnMinX-factorX)*mfGridElementWidthInv)); if(nMinCellX>=mnGridCols) return vIndices; // ceil向上取整 const int nMaxCellX = min((int)mnGridCols-1,(int)ceil((x-mnMinX+factorX)*mfGridElementWidthInv)); if(nMaxCellX<0) return vIndices; const int nMinCellY = max(0,(int)floor((y-mnMinY-factorY)*mfGridElementHeightInv)); if(nMinCellY>=mnGridRows) return vIndices; const int nMaxCellY = min((int)mnGridRows-1,(int)ceil((y-mnMinY+factorY)*mfGridElementHeightInv)); if(nMaxCellY<0) return vIndices; // 遍历每个cell,取出其中每个cell中的点,并且每个点都要计算是否在邻域内 for(int ix = nMinCellX; ix<=nMaxCellX; ix++) { for(int iy = nMinCellY; iy<=nMaxCellY; iy++) { const vector vCell = (!bRight) ? mGrid[ix][iy] : mGridRight[ix][iy]; for(size_t j=0, jend=vCell.size(); j=mnMinX && x=mnMinY && y0) { // 由2维图像反投影到相机坐标系 // 双目中mvDepth是在ComputeStereoMatches函数中求取的,rgbd中是直接测量的 const float u = mvKeys[i].pt.x; const float v = mvKeys[i].pt.y; const float x = (u-cx)*z*invfx; const float y = (v-cy)*z*invfy; cv::Mat x3Dc = (cv::Mat_(3,1) << x, y, z); unique_lock lock(mMutexPose); // 由相机坐标系转换到世界坐标系 // Twc为相机坐标系到世界坐标系的变换矩阵 // Twc.rosRange(0,3).colRange(0,3)取Twc矩阵的前3行与前3列 return Twc.rowRange(0,3).colRange(0,3)*x3Dc+Twc.rowRange(0,3).col(3); } else return cv::Mat(); } // Compute Scene Depth (q=2 median). Used in monocular. 评估当前关键帧场景深度,q=2表示中值. 只是在单目情况下才会使用 // 其实过程就是对当前关键帧下所有地图点的深度进行从小到大排序,返回距离头部其中1/q处的深度值作为当前场景的平均深度 float KeyFrame::ComputeSceneMedianDepth(const int q) { vector vpMapPoints; cv::Mat Tcw_; { unique_lock lock(mMutexFeatures); unique_lock lock2(mMutexPose); vpMapPoints = mvpMapPoints; Tcw_ = Tcw.clone(); } vector vDepths; vDepths.reserve(N); cv::Mat Rcw2 = Tcw_.row(2).colRange(0,3); Rcw2 = Rcw2.t(); float zcw = Tcw_.at(2,3); // 遍历每一个地图点,计算并保存其在当前关键帧下的深度 for(int i=0; iGetWorldPos(); float z = Rcw2.dot(x3Dw)+zcw; // (R*x3Dw+t)的第三行,即z vDepths.push_back(z); } } sort(vDepths.begin(),vDepths.end()); return vDepths[(vDepths.size()-1)/q]; } void KeyFrame::SetNewBias(const IMU::Bias &b) { unique_lock lock(mMutexPose); mImuBias = b; if(mpImuPreintegrated) mpImuPreintegrated->SetNewBias(b); } cv::Mat KeyFrame::GetGyroBias() { unique_lock lock(mMutexPose); return (cv::Mat_(3,1) << mImuBias.bwx, mImuBias.bwy, mImuBias.bwz); } cv::Mat KeyFrame::GetAccBias() { unique_lock lock(mMutexPose); return (cv::Mat_(3,1) << mImuBias.bax, mImuBias.bay, mImuBias.baz); } IMU::Bias KeyFrame::GetImuBias() { unique_lock lock(mMutexPose); return mImuBias; } Map* KeyFrame::GetMap() { unique_lock lock(mMutexMap); return mpMap; } void KeyFrame::UpdateMap(Map* pMap) { unique_lock lock(mMutexMap); mpMap = pMap; } void KeyFrame::PreSave(set& spKF,set& spMP, set& spCam) { // Save the id of each MapPoint in this KF, there can be null pointer in the vector mvBackupMapPointsId.clear(); mvBackupMapPointsId.reserve(N); for(int i = 0; i < N; ++i) { if(mvpMapPoints[i] && spMP.find(mvpMapPoints[i]) != spMP.end()) // Checks if the element is not null mvBackupMapPointsId.push_back(mvpMapPoints[i]->mnId); else // If the element is null his value is -1 because all the id are positives mvBackupMapPointsId.push_back(-1); } //cout << "KeyFrame: ID from MapPoints stored" << endl; // Save the id of each connected KF with it weight mBackupConnectedKeyFrameIdWeights.clear(); for(std::map::const_iterator it = mConnectedKeyFrameWeights.begin(), end = mConnectedKeyFrameWeights.end(); it != end; ++it) { if(spKF.find(it->first) != spKF.end()) mBackupConnectedKeyFrameIdWeights[it->first->mnId] = it->second; } //cout << "KeyFrame: ID from connected KFs stored" << endl; // Save the parent id mBackupParentId = -1; if(mpParent && spKF.find(mpParent) != spKF.end()) mBackupParentId = mpParent->mnId; //cout << "KeyFrame: ID from Parent KF stored" << endl; // Save the id of the childrens KF mvBackupChildrensId.clear(); mvBackupChildrensId.reserve(mspChildrens.size()); for(KeyFrame* pKFi : mspChildrens) { if(spKF.find(pKFi) != spKF.end()) mvBackupChildrensId.push_back(pKFi->mnId); } //cout << "KeyFrame: ID from Children KFs stored" << endl; // Save the id of the loop edge KF mvBackupLoopEdgesId.clear(); mvBackupLoopEdgesId.reserve(mspLoopEdges.size()); for(KeyFrame* pKFi : mspLoopEdges) { if(spKF.find(pKFi) != spKF.end()) mvBackupLoopEdgesId.push_back(pKFi->mnId); } //cout << "KeyFrame: ID from Loop KFs stored" << endl; // Save the id of the merge edge KF mvBackupMergeEdgesId.clear(); mvBackupMergeEdgesId.reserve(mspMergeEdges.size()); for(KeyFrame* pKFi : mspMergeEdges) { if(spKF.find(pKFi) != spKF.end()) mvBackupMergeEdgesId.push_back(pKFi->mnId); } //cout << "KeyFrame: ID from Merge KFs stored" << endl; //Camera data mnBackupIdCamera = -1; if(mpCamera && spCam.find(mpCamera) != spCam.end()) mnBackupIdCamera = mpCamera->GetId(); //cout << "KeyFrame: ID from Camera1 stored; " << mnBackupIdCamera << endl; mnBackupIdCamera2 = -1; if(mpCamera2 && spCam.find(mpCamera2) != spCam.end()) mnBackupIdCamera2 = mpCamera2->GetId(); //cout << "KeyFrame: ID from Camera2 stored; " << mnBackupIdCamera2 << endl; //Inertial data mBackupPrevKFId = -1; if(mPrevKF && spKF.find(mPrevKF) != spKF.end()) mBackupPrevKFId = mPrevKF->mnId; //cout << "KeyFrame: ID from Prev KF stored" << endl; mBackupNextKFId = -1; if(mNextKF && spKF.find(mNextKF) != spKF.end()) mBackupNextKFId = mNextKF->mnId; //cout << "KeyFrame: ID from NextKF stored" << endl; if(mpImuPreintegrated) mBackupImuPreintegrated.CopyFrom(mpImuPreintegrated); //cout << "KeyFrame: Imu Preintegrated stored" << endl; } void KeyFrame::PostLoad(map& mpKFid, map& mpMPid, map& mpCamId){ // Rebuild the empty variables // Pose SetPose(Tcw); // Reference reconstruction // Each MapPoint sight from this KeyFrame mvpMapPoints.clear(); mvpMapPoints.resize(N); for(int i=0; i(NULL); } // Conected KeyFrames with him weight mConnectedKeyFrameWeights.clear(); for(map::const_iterator it = mBackupConnectedKeyFrameIdWeights.begin(), end = mBackupConnectedKeyFrameIdWeights.end(); it != end; ++it) { KeyFrame* pKFi = mpKFid[it->first]; mConnectedKeyFrameWeights[pKFi] = it->second; } // Restore parent KeyFrame if(mBackupParentId>=0) mpParent = mpKFid[mBackupParentId]; // KeyFrame childrens mspChildrens.clear(); for(vector::const_iterator it = mvBackupChildrensId.begin(), end = mvBackupChildrensId.end(); it!=end; ++it) { mspChildrens.insert(mpKFid[*it]); } // Loop edge KeyFrame mspLoopEdges.clear(); for(vector::const_iterator it = mvBackupLoopEdgesId.begin(), end = mvBackupLoopEdgesId.end(); it != end; ++it) { mspLoopEdges.insert(mpKFid[*it]); } // Merge edge KeyFrame mspMergeEdges.clear(); for(vector::const_iterator it = mvBackupMergeEdgesId.begin(), end = mvBackupMergeEdgesId.end(); it != end; ++it) { mspMergeEdges.insert(mpKFid[*it]); } //Camera data if(mnBackupIdCamera >= 0) { mpCamera = mpCamId[mnBackupIdCamera]; } if(mnBackupIdCamera2 >= 0) { mpCamera2 = mpCamId[mnBackupIdCamera2]; } //Inertial data if(mBackupPrevKFId != -1) { mPrevKF = mpKFid[mBackupPrevKFId]; } if(mBackupNextKFId != -1) { mNextKF = mpKFid[mBackupNextKFId]; } mpImuPreintegrated = &mBackupImuPreintegrated; // Remove all backup container mvBackupMapPointsId.clear(); mBackupConnectedKeyFrameIdWeights.clear(); mvBackupChildrensId.clear(); mvBackupLoopEdgesId.clear(); UpdateBestCovisibles(); //ComputeSceneMedianDepth(); } bool KeyFrame::ProjectPointDistort(MapPoint* pMP, cv::Point2f &kp, float &u, float &v) { // 3D in absolute coordinates cv::Mat P = pMP->GetWorldPos(); cv::Mat Rcw = Tcw.rowRange(0, 3).colRange(0, 3); cv::Mat tcw = Tcw.rowRange(0, 3).col(3); // 3D in camera coordinates cv::Mat Pc = Rcw*P+tcw; float &PcX = Pc.at(0); float &PcY= Pc.at(1); float &PcZ = Pc.at(2); // Check positive depth if(PcZ<0.0f) { cout << "Negative depth: " << PcZ << endl; return false; } // Project in image and check it is not outside float invz = 1.0f/PcZ; u=fx*PcX*invz+cx; v=fy*PcY*invz+cy; // cout << "c"; if(umnMaxX) return false; if(vmnMaxY) return false; float x = (u - cx) * invfx; float y = (v - cy) * invfy; float r2 = x * x + y * y; float k1 = mDistCoef.at(0); float k2 = mDistCoef.at(1); float p1 = mDistCoef.at(2); float p2 = mDistCoef.at(3); float k3 = 0; if(mDistCoef.total() == 5) { k3 = mDistCoef.at(4); } // Radial distorsion float x_distort = x * (1 + k1 * r2 + k2 * r2 * r2 + k3 * r2 * r2 * r2); float y_distort = y * (1 + k1 * r2 + k2 * r2 * r2 + k3 * r2 * r2 * r2); // Tangential distorsion x_distort = x_distort + (2 * p1 * x * y + p2 * (r2 + 2 * x * x)); y_distort = y_distort + (p1 * (r2 + 2 * y * y) + 2 * p2 * x * y); float u_distort = x_distort * fx + cx; float v_distort = y_distort * fy + cy; u = u_distort; v = v_distort; kp = cv::Point2f(u, v); return true; } bool KeyFrame::ProjectPointUnDistort(MapPoint* pMP, cv::Point2f &kp, float &u, float &v) { // 3D in absolute coordinates cv::Mat P = pMP->GetWorldPos(); cv::Mat Rcw = Tcw.rowRange(0, 3).colRange(0, 3); cv::Mat tcw = Tcw.rowRange(0, 3).col(3); // 3D in camera coordinates cv::Mat Pc = Rcw*P+tcw; float &PcX = Pc.at(0); float &PcY= Pc.at(1); float &PcZ = Pc.at(2); // Check positive depth if(PcZ<0.0f) { cout << "Negative depth: " << PcZ << endl; return false; } // Project in image and check it is not outside const float invz = 1.0f/PcZ; u=fx*PcX*invz+cx; v=fy*PcY*invz+cy; // cout << "c"; if(umnMaxX) return false; if(vmnMaxY) return false; kp = cv::Point2f(u, v); return true; } cv::Mat KeyFrame::GetRightPose() { unique_lock lock(mMutexPose); cv::Mat Rrl = mTlr.rowRange(0,3).colRange(0,3).t(); cv::Mat Rlw = Tcw.rowRange(0,3).colRange(0,3).clone(); cv::Mat Rrw = Rrl * Rlw; cv::Mat tlw = Tcw.rowRange(0,3).col(3).clone(); cv::Mat trl = - Rrl * mTlr.rowRange(0,3).col(3); cv::Mat trw = Rrl * tlw + trl; cv::Mat Trw; cv::hconcat(Rrw,trw,Trw); return Trw.clone(); } cv::Mat KeyFrame::GetRightPoseInverse() { unique_lock lock(mMutexPose); cv::Mat Rrl = mTlr.rowRange(0,3).colRange(0,3).t(); cv::Mat Rlw = Tcw.rowRange(0,3).colRange(0,3).clone(); cv::Mat Rwr = (Rrl * Rlw).t(); cv::Mat Rwl = Tcw.rowRange(0,3).colRange(0,3).t(); cv::Mat tlr = mTlr.rowRange(0,3).col(3); cv::Mat twl = GetCameraCenter(); cv::Mat twr = Rwl * tlr + twl; cv::Mat Twr; cv::hconcat(Rwr,twr,Twr); return Twr.clone(); } cv::Mat KeyFrame::GetRightPoseInverseH() { unique_lock lock(mMutexPose); cv::Mat Rrl = mTlr.rowRange(0,3).colRange(0,3).t(); cv::Mat Rlw = Tcw.rowRange(0,3).colRange(0,3).clone(); cv::Mat Rwr = (Rrl * Rlw).t(); cv::Mat Rwl = Tcw.rowRange(0,3).colRange(0,3).t(); cv::Mat tlr = mTlr.rowRange(0,3).col(3); cv::Mat twl = Ow.clone(); cv::Mat twr = Rwl * tlr + twl; cv::Mat Twr; cv::hconcat(Rwr,twr,Twr); cv::Mat h(1,4,CV_32F,cv::Scalar(0.0f)); h.at(3) = 1.0f; cv::vconcat(Twr,h,Twr); return Twr.clone(); } cv::Mat KeyFrame::GetRightCameraCenter() { unique_lock lock(mMutexPose); cv::Mat Rwl = Tcw.rowRange(0,3).colRange(0,3).t(); cv::Mat tlr = mTlr.rowRange(0,3).col(3); cv::Mat twl = Ow.clone(); cv::Mat twr = Rwl * tlr + twl; return twr.clone(); } cv::Mat KeyFrame::GetRightRotation() { unique_lock lock(mMutexPose); cv::Mat Rrl = mTlr.rowRange(0,3).colRange(0,3).t(); cv::Mat Rlw = Tcw.rowRange(0,3).colRange(0,3).clone(); cv::Mat Rrw = Rrl * Rlw; return Rrw.clone(); } cv::Mat KeyFrame::GetRightTranslation() { unique_lock lock(mMutexPose); cv::Mat Rrl = mTlr.rowRange(0,3).colRange(0,3).t(); cv::Mat tlw = Tcw.rowRange(0,3).col(3).clone(); cv::Mat trl = - Rrl * mTlr.rowRange(0,3).col(3); cv::Mat trw = Rrl * tlw + trl; return trw.clone(); } void KeyFrame::SetORBVocabulary(ORBVocabulary* pORBVoc) { mpORBvocabulary = pORBVoc; } void KeyFrame::SetKeyFrameDatabase(KeyFrameDatabase* pKFDB) { mpKeyFrameDB = pKFDB; } } //namespace ORB_SLAM