我们可以使用下面的代码来实现图文检索:import torch import torchvision# 加载数据集 transform = torchvision.transforms.Compose([ torchvision.transforms.Resize(224), torchvision.transforms.ToTensor() ])dataset = torchvision.datasets.ImageNet(root='./data/', split='train', trans
在pytorch框架下写一段图文检索的代码
最新推荐文章于 2024-08-17 22:31:43 发布