1 前言

今天在学习PyTorch~

在进行训练时,有一个需求就是,我们需要用一个整形的数字作为样本图片的唯一ID,我想到的方法就是使用Hash算法对图片的内部路径进行编码~

2 在Python中获取图片的唯一编码

3 将编码转换为torch.Tensor

我曾经试过将获得的Hash编码转成IntTensor,不过没有成功,因为用hashlib生成的Hash值太长了,用IntTensor类型无法存下,

我想我们可以使用Int64Tensor来试一下,看是否可以成功保存;

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐