2017-11-17 3 views
0

Wie kann ich Bounding Box und nicht die normalisierte Koordinate in tensorflow object-detection api verwenden? in object_detection_tutorial.ipynb, bemerkte ich, dass die Standardkoordinate in Normalize-Koordinaten ist und die Form der Boxen [xmin, ymin, xmax, ymax] ist und wie man sie in [image_length xmin, image_width ymin, image_length xmax, image_width ymax ]? Ich versucheWie kann ich Bounding-Box zeichnen und nicht die normalisierte Koordinate in Tensorflow Objekterkennung api

 boxes[0]=boxes[0]*200 
     boxes[1]=boxes[1]*100 
     boxes[2]=boxes[2]*200 
     boxes[3]=boxes[3]*100 

zu verwenden, aber Fehler passiert:

--------------------------------------------------------------------------- 
IndexError        Traceback (most recent call last) 
<ipython-input-72-efcec9615ee3> in <module>() 
    30     feed_dict={image_tensor: image_np_expanded}) 
    31     boxes[0]=boxes[0]*200 
---> 32     boxes[1]=boxes[1]*100 
    33     boxes[2]=boxes[2]*200 
    34     boxes[3]=boxes[3]*100 
IndexError: index 1 is out of bounds for axis 0 with size 1 
+0

Überprüfen Sie die Dims der Boxen Variable. Bei Index Null sind die Bboxen für Bild eins. Bei Index 1 ist die Nx4-Matrix von B-Boxen für das zweite Bild, und so weiter ... –

Antwort

0

, wenn Sie bei Forschung aussehen/object_detection/utils/visualization_utils.py die Boxen [0] yMin nicht xMin, wenn Sie diese Koordinaten multiplizieren mit 100 oder 200 stellen Sie sicher, dass es immer noch in der Bildgrenze ist (im_width, im_height).

Sie können versuchen, Boxen [0] * 100, Boxen [1] * -200, Boxen [2] * -100, Boxen [3] * 200, das ist ähnlich zu diesem Code.

ymin = boxes[0]*100 
xmin = boxes[1]*-200 
ymax = boxes[2]*-100 
xmax = boxes[3]*200 

draw = ImageDraw.Draw(image) 
im_width, im_height = image.size 
(left, right, top, bottom) = (xmin * im_width, xmax * im_width, 
           ymin * im_height, ymax * im_height) 

draw.line([(left, top), (left, bottom), (right, bottom), 
       (right, top), (left, top)], width=thickness, fill=color) 
+0

vielen Dank, es funktioniert! –

Verwandte Themen