AndsGo / reverse_image_search

以图搜图基于Towhee(resnet50 模型) + Milvus
41 stars 11 forks source link

resnet50模型本地调用 #8

Open liyunfei0302 opened 6 months ago

liyunfei0302 commented 6 months ago

您这个是调用的在线模型吧,离线环境可以使用吗,或者说把resnet50下载到本地,调用本地模型

AndsGo commented 6 months ago

第一次使用会下载模型,下载完后面就使用的是本地模型了

离线的使用,可以试着手动将模型复制到 linux 环境: /root/.cache/torch/hub/checkpoints ,windows 默认下载到 C:\Users\Administrator.cache\torch\hub\checkpoints

模型地址 : https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-rsb-weights/resnet50_a1_0-14fe96d1.pth