harshtable 不明白之处

在data_decoders/tf_example_decoder.py
if label_map_proto_file:
  label_map = label_map_util.get_label_map_dict(label_map_proto_file,
                                                use_display_name)
  # We use a default_value of -1, but we expect all labels to be contained
  # in the label map.
  table = tf.contrib.lookup.HashTable(
      initializer=tf.contrib.lookup.KeyValueTensorInitializer(
          keys=tf.constant(list(label_map.keys())),
          values=tf.constant(list(label_map.values()), dtype=tf.int64)),
      default_value=-1)
  # If the label_map_proto is provided, try to use it in conjunction with
  # the class text, and fall back to a materialized ID.
  # TODO(lzc): note that here we are using BackupHandler defined in this
  # file(which is branching slim_example_decoder.BackupHandler). Need to
  # switch back to slim_example_decoder.BackupHandler once tf 1.5 becomes
  # more popular.
  label_handler = BackupHandler(
      LookupTensor('image/object/class/text', table, default_value=''),
      slim_example_decoder.Tensor('image/object/class/label'))
else:
  label_handler = slim_example_decoder.Tensor('image/object/class/label')
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值