0

我正在尝试在本地计算机上运行 deeplab。我在我的电脑上安装了 deeplab 并定义了路径。Deeplab 在我的本地计算机上成功运行。这是 deeplab_demo.ipynb 的最后一个块

SAMPLE_IMAGE = 'image1'  # @param ['image1', 'image2', 'image3']
IMAGE_URL = ''  #@param {type:"string"}

_SAMPLE_URL = ('https://github.com/tensorflow/models/blob/master/research/'
               'deeplab/g3doc/img/%s.jpg?raw=true')


def run_visualization(url):
  """Inferences DeepLab model and visualizes result."""
  try:
    f = urllib.request.urlopen(url)
    jpeg_str = f.read()
    original_im = Image.open(BytesIO(jpeg_str))
  except IOError:
    print('Cannot retrieve image. Please check url: ' + url)
    return

  print('running deeplab on image %s...' % url)
  resized_im, seg_map = MODEL.run(original_im)

  vis_segmentation(resized_im, seg_map)


image_url = IMAGE_URL or _SAMPLE_URL % SAMPLE_IMAGE
run_visualization(image_url)

我正在运行它 jupyter notebook。它适用于“_SAMPLE_URL”变量或任何包含图像文件的链接。我想用本地图像文件测试 deeplab。

注意：环境变量已定义。所以我可以从这个笔记本访问我的本地文件。所有图书馆都在工作。但我不想在 URL 上测试图像，只是本地图像文件。

4

2 回答 2

1

更改本地路径

_SAMPLE_URL = ('https://github.com/tensorflow/models/blob/master/research/''deeplab/g3doc/img/%s.jpg?raw=true')

--->

_SAMPLE_URL = ('file:///path to/tensorflow/models/blob/master/research/' 'deeplab/g3doc/img/%s.jpg')

于 2021-07-21T00:10:52.407 回答
0

我通过在 'original_im = Image.open(BytesIO(jpeg_str))' 行将 'BytesIO(jpeg_str)' 更改为 '123.jpg' 来做到这一点。

于 2021-03-03T21:42:49.173 回答