Python高维数组排序,根据一个array中的值将另一个array排序

最近在做CBIR的时候碰到了这样一个问题: 给定一张图片,在数据库中检索和输入图片相似的图片。比如给定n张图片作为输入,输出数据库中各自top k检索结果。然后需要将这些图片分别再根据特征相似度得分进行重排。例:四张图片的top 3检索结果的index为 1: 23,34,32 2: 93,23,21 3: 52,12,13 4: 37,62,10 对应的局部特征相似度为: 1:0.3, 0.8, 0.1 2:0.4, 0.2, 0.8 3:0.1, 0.9, 0.7 4:0.6, 0.4, 0.8 需要分别将四张输入图片各自检索出的图片index根据得分进行重排,即输出应为: 1: 34,23,32 2: 21,93,23 3: 12,13,52 4: 10,37,62 传统方法,用for循环,数据量一大就很费时。想用numba加速,又需要避免跳出numpy模块。来看代码
import numpy as n

Python高维数组排序,根据一个array中的值将另一个array排序最先出现在Python成神之路

版权声明:
作者:dingding
链接:https://www.techfm.club/p/8020.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>