4 个月前

PIGEON:预测图像地理位置

PIGEON:预测图像地理位置

摘要

全球规模的图像地理定位仍然是一个具有挑战性的问题,因为这些图像可能来自世界各地,具有极大的多样性。尽管基于视觉变换器的方法在地理定位精度方面取得了显著进展,但先前文献中的成功仅限于地标图像的狭窄分布,其性能尚未推广到未见过的地方。我们提出了一种新的地理定位系统,该系统结合了语义地理单元创建、多任务对比预训练以及一种新颖的损失函数。此外,我们的工作首次在位置聚类上进行检索以改进猜测结果。我们在街景数据和通用图像地理定位两个方面对两个模型进行了评估;第一个模型称为PIGEON(基于Geoguessr游戏的数据训练),能够在全球范围内将其超过40%的猜测放置在目标位置25公里以内。我们还开发了一个机器人,并在一个盲测实验中将PIGEON与人类玩家进行对抗,最终排名在所有玩家的前0.01%。我们进一步挑战了世界上顶尖的职业Geoguessr玩家进行六场对决,每场比赛都有数百万观众观看,结果我们赢得了全部六场比赛。第二个模型称为PIGEOTTO(基于Flickr和Wikipedia图像数据集训练),在广泛的图像地理定位基准测试中取得了最先进的成果,在城市级别准确度上比之前的最佳方法高出7.7个百分点,在国家级别准确度上则高出38.8个百分点。我们的研究结果表明,PIGEOTTO是首个能够有效推广到未见过地方的图像地理定位模型,并且我们的方法可以为高精度、全球规模的图像地理定位系统铺平道路。我们的代码已在GitHub上公开。

代码仓库

LukasHaas/PIGEON
官方
pytorch

基准测试

基准方法指标
photo-geolocation-estimation-on-gws15kPIGEOTTO
City level (25 km): 9.2
Continent level (2500 km): 85.1
Country level (750 km): 65.7
Median Error (km): 415.4
Region level (200 km): 31.2
Street level (1 km): 0.7
photo-geolocation-estimation-on-im2gpsPIGEOTTO
City level (25 km): 40.9
Continent level (2500 km): 91.1
Country level (750 km): 82.3
Median Error (km): 70.5
Reference images: 4.5M
Region level (200 km): 63.3
Street level (1 km): 14.8
Training images: 4.5M
photo-geolocation-estimation-on-im2gps3kPIGEOTTO
City level (25 km): 36.7
Continent level (2500 km): 85.3
Country level (750 km): 72.4
Median Error (km): 147.3
Region level (200 km): 53.8
Street level (1 km): 11.3
Training Images: 4.5M
photo-geolocation-estimation-on-yfcc26kPIGEOTTO
City level (25 km): 25.8
Continent level (2500 km): 79.0
Country level (750 km): 63.2
Median Error (km): 333.3
Region level (200 km): 42.7
Street level (1 km): 10.5
Training Images: 4.5M
photo-geolocation-estimation-on-yfcc4kPIGEOTTO
City (25 km): 23.7
Continent (2500 km): 77.7
Country (750 km): 62.2
Median Error (km): 383.0
Region (200 km): 40.6
Street (1 km): 10.4

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
PIGEON:预测图像地理位置 | 论文 | HyperAI超神经