行人重识别的小技巧利用shuffle函数随机排序后随机快速得到一个同id的pos对

首先介绍一下shuffle()函数,很简单就是将序列中的所有元素随机排序:如下所示

import random
a=list(range(6))
random.shuffle(a)
print(a)

随机排序后输出为[1, 5, 4, 0, 3, 2],接着如下简单的使用,在ReID中常用的代码块,随机排序后随机快速的遍历到与该索引号对应行人相同的id。

def serch_pos(self, index):
   path, target, cam = self.imgs[index]
   img = self.loader(path)
   if self.true_pair:#找到和该图片一个行人id的,不一定是同一张图片,即pos对
       img = self.transform(img)
       random_index = list(range(self.len))
       random.shuffle(random_index)#将序列中的所有元素随机排序
       for i in random_index:
           tpath, ttarget, tcam = self.imgs[i]
           if ttarget == target:
                timg = self.loader(tpath)
                timg = self.transform(timg)
                return img, target, path, cam, timg, tcam
#随机排序后遍历到同一行人id标签的图像 ,返回图像,personID,路径,cam id,以及同id变换后的图像和cam id
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值