當前位置: 首頁>>代碼示例>>Python>>正文


Python common.TEST_SET屬性代碼示例

本文整理匯總了Python中deeplab.common.TEST_SET屬性的典型用法代碼示例。如果您正苦於以下問題:Python common.TEST_SET屬性的具體用法?Python common.TEST_SET怎麽用?Python common.TEST_SET使用的例子?那麽, 這裏精選的屬性代碼示例或許可以為您提供幫助。您也可以進一步了解該屬性所在deeplab.common的用法示例。


在下文中一共展示了common.TEST_SET屬性的1個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Python代碼示例。

示例1: _get_data

# 需要導入模塊: from deeplab import common [as 別名]
# 或者: from deeplab.common import TEST_SET [as 別名]
def _get_data(data_provider, dataset_split):
  """Gets data from data provider.

  Args:
    data_provider: An object of slim.data_provider.
    dataset_split: Dataset split.

  Returns:
    image: Image Tensor.
    label: Label Tensor storing segmentation annotations.
    image_name: Image name.
    height: Image height.
    width: Image width.

  Raises:
    ValueError: Failed to find label.
  """
  if common.LABELS_CLASS not in data_provider.list_items():
    raise ValueError('Failed to find labels.')

  image, height, width = data_provider.get(
      [common.IMAGE, common.HEIGHT, common.WIDTH])

  # Some datasets do not contain image_name.
  if common.IMAGE_NAME in data_provider.list_items():
    image_name, = data_provider.get([common.IMAGE_NAME])
  else:
    image_name = tf.constant('')

  label = None
  if dataset_split != common.TEST_SET:
    label, = data_provider.get([common.LABELS_CLASS])

  return image, label, image_name, height, width 
開發者ID:itsamitgoel,項目名稱:Gun-Detector,代碼行數:36,代碼來源:input_generator.py


注:本文中的deeplab.common.TEST_SET屬性示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台,相關代碼片段篩選自各路編程大神貢獻的開源項目,源碼版權歸原作者所有,傳播和使用請參考對應項目的License;未經允許,請勿轉載。