-
大小: 7.11MB文件類型: .zip金幣: 2下載: 0 次發布日期: 2023-09-23
- 語言: Python
- 標簽: superpoint??圖像拼接??python??deeplearning??
資源簡介
博客連接https://blog.csdn.net/qq_33591712/article/details/84947829#comments
用superpoint方法代替surf提取圖像特征,進行Python版本的圖像拼接。

代碼片段和文件信息
import?cv2
import?numpy?as?np?
from?sp_extractor?import?SuperPointFrontend
class?matchers:
def?__init__(self):
self.surf?=?cv2.xfeatures2d.SURF_create()
????????????????self.detector?=?SuperPointFrontend(weights_path=“superpoint_v1.pth“
???????????????????????????????????????????nms_dist=4
???????????????????????????????????????????conf_thresh=0.015
???????????????????????????????????????????nn_thresh=0.7
???????????????????????????????????????????cuda=True)
FLANN_INDEX_KDTREE?=?0
index_params?=?dict(algorithm=0?trees=5)
search_params?=?dict(checks=50)
self.flann?=?cv2.FlannbasedMatcher(index_params?search_params)
def?match(self?i1?i2?direction=None):
imageSet1?=?self.getSURFFeatures(i1)
imageSet2?=?self.getSURFFeatures(i2)
print?“Direction?:?“?direction
????????????????matches?=?self.flann.knnMatch(
np.asarray(imageSet2[‘des‘]np.float32)
np.asarray(imageSet1[‘des‘]np.float32)
k=2
)
good?=?[]
for?i??(m?n)?in?enumerate(matches):
if?m.distance?0.7*n.distance:
good.append((m.trainIdx?m.queryIdx))
if?len(good)?>?4:
pointsCurrent?=?imageSet2[‘kp‘]
pointsPrevious?=?imageSet1[‘kp‘]
matchedPointsCurrent?=?np.float32(
[pointsCurrent[i]?for?(__?i)?in?good]
)
matchedPointsPrev?=?np.float32(
[pointsPrevious[i]?for?(i?__)?in?good]
)
H?s?=?cv2.findHomography(matchedPointsCurrent?matchedPointsPrev?cv2.RANSAC?4)
return?H
return?None
def?getSURFFeatures(self?im):
gray?=?cv2.cvtColor(im?cv2.COLOR_BGR2GRAY)
????????????????pts?desc?heatmap?=?self.detector.run(gray)
#kp?des?=?self.surf.detectAndCompute(gray?None)
????????????????pts=np.delete(pts2axis=0)
????????????????desc=np.delete(desc2axis=0)
????????????????pts=np.transpose(pts)
????????????????desc=np.transpose(desc)
????????????????pts=pts.tolist()
????????????????desc=desc.tolist()
????????????????
????????????????return?{‘kp‘:pts?‘des‘:desc}
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2018-11-11?19:38??Python-Multiple-Image-Stitching\
?????目錄???????????0??2019-01-06?21:34??Python-Multiple-Image-Stitching\.git\
?????文件??????????23??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\HEAD
?????目錄???????????0??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\branches\
?????文件?????????290??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\config
?????文件??????????73??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\desc
?????目錄???????????0??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\hooks\
?????文件?????????478??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\hooks\applypatch-msg.sample
?????文件?????????896??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\hooks\commit-msg.sample
?????文件?????????189??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\hooks\post-update.sample
?????文件?????????424??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\hooks\pre-applypatch.sample
?????文件????????1642??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\hooks\pre-commit.sample
?????文件????????1348??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\hooks\pre-push.sample
?????文件????????4898??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\hooks\pre-reba
?????文件????????1239??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\hooks\prepare-commit-msg.sample
?????文件????????3610??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\hooks\update.sample
?????文件????????2248??2018-11-13?21:33??Python-Multiple-Image-Stitching\.git\index
?????目錄???????????0??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\info\
?????文件?????????240??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\info\exclude
?????目錄???????????0??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\logs\
?????文件?????????219??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\logs\HEAD
?????目錄???????????0??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\logs\refs\
?????目錄???????????0??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\logs\refs\heads\
?????文件?????????219??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\logs\refs\heads\master
?????目錄???????????0??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\logs\refs\remotes\
?????目錄???????????0??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\logs\refs\remotes\origin\
?????文件?????????219??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\logs\refs\remotes\origin\HEAD
?????目錄???????????0??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\ob
?????目錄???????????0??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\ob
?????文件?????????116??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\ob
?????目錄???????????0??2018-11-11?19:22??Python-Multiple-Image-Stitching\.git\ob
............此處省略198個文件信息
評論
共有 條評論