当前位置:   article > 正文

由于网络问题,无法获取ImageNet对应的1000类的类别标签的解决办法_如何获取imagenet用于图像分类的1000个标签

如何获取imagenet用于图像分类的1000个标签

这次使用已经预训练好的VGG16网络+ImageNet中的json格式的label,对上传的图片进行类型预测。
具体的代码可以点击这里

但是,在导入json数据的时候,我遇到无法显示json数据的问题。
源码如下所示:

import requests
## 获取vgg模型训练时对应的1000类的类别标签
LABELS_URL = "https://s3.amazonaws.com/outcome-blog/imagenet/labels.json"
# https://s3.amazonaws.com/outcome-blog/imagenet/labels.json
# 从网页链接中获取类别标签
response = requests.get(LABELS_URL)  
labels = {int(key): value for key, value in response.json().items()}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7

于是,我点开 LABEL_URL对应的网址,果然是有问题的。
网页长这个样子:在这里插入图片描述

解决方法:我们需要手动下载ImageNet的labels(json)格式的对象,(点击获取百度网盘链接,密码是:1111)
如果无法通过百度云下载,请自行到这个网址下载:点我!
下载后,我们把json格式的文件放在工作路径下。
在这里插入图片描述
然后,再使用如下代码:

import json

jsonfile = r'labels.json'   # 换成你们自己的地址,注意前面的r不要删
with open(jsonfile, 'r') as load_f:
    load_json = json.load(load_f)
 
 
# 我的类别名词叫做labels,根据你们自己的类别名可以进行修改
labels = {int(key): value for (key, value)
          in load_json.items()}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/凡人多烦事01/article/detail/340912
推荐阅读
  

闽ICP备14008679号