From 1c66012ac9ccca6c9e8a3dcf24f737c1a0f734c7 Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Sun, 19 Aug 2018 19:29:35 -0700 Subject: [PATCH] fix import (#1621) --- nnvm/python/nnvm/testing/yolo2_detection.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/nnvm/python/nnvm/testing/yolo2_detection.py b/nnvm/python/nnvm/testing/yolo2_detection.py index b7744c45..0b229149 100644 --- a/nnvm/python/nnvm/testing/yolo2_detection.py +++ b/nnvm/python/nnvm/testing/yolo2_detection.py @@ -10,9 +10,6 @@ from __future__ import division import math from collections import namedtuple import numpy as np -from PIL import Image -from PIL import ImageDraw -from PIL import ImageFont def _entry_index(batch, w, h, outputs, classes, coords, location, entry): n = int(location/(w*h)) @@ -186,6 +183,10 @@ def _draw_label(im, r, c, label, rgb): _set_pixel(im, i+c, j+r, k, val)#rgb[k] * val) def _get_label(labelstr, rgb): + from PIL import Image + from PIL import ImageDraw + from PIL import ImageFont + text = labelstr colorText = "black" testDraw = ImageDraw.Draw(Image.new('RGB', (1, 1)))