Search code examples
pythonimagepython-2.7drawdrawrectangle

how to draw rectangles using list in python


for line, images_files in zip(lines, image_list):
                    info = line.split(',')

                    image_index = [int(info[0])]
                    box_coordiante1 = [info[2]]
                    box_coordiante2 = [info[3]]
                    box_coordiante3 = [info[4]]
                    box_coordiante4 = [info[5]]

                    prev_image_num = 1
                    for image_number in image_index:    #### read each other image_number

                        if prev_image_num != image_number: # if read 11111 but appear different number such as 2, 3 and ect
                            prev_image_num = image_number  # the different number becomes pre_image_num(it was 1)
                            #box_coordinate = []  # empty box_coordinate
                            #box_coordinate.append(info[2:6])
                            #print box_coordinate
                            # box_coordinate.append() #insert 2 to 6 axis
                        rect = plt.Rectangle((int(box_coordiante1), int(box_coordiante2)), int(box_coordiante3), int(box_coordiante4), linewidth=1, edgecolor='r', facecolor='none')
                        ax.add_patch(rect)

                    im = cv2.imread(images_files)
                    im = im[:, :, (2, 1, 0)]
                    #     # Display the image
                    plt.imshow(im)
                    plt.draw()
                    plt.pause(0.1)
                    plt.cla()

I am supposed to draw boxes on each picture. For showing boxes on each picture, i guess that gather location of boxes and show them at that same time. So i used a way using LIST to plt.Rectanle but it said "TypeError: int() argument must be a string or a number, not 'list'" Are there other ways??


Solution

  • I'm not very familiar with Python, but it seems like you want a plain number in the variables image_index and box_coordinateN. It looks like you're assigning single-element arrays to them. Try changing:

    image_index = [int(info[0])] // list containing one element: int(info[0])
    box_coordiante1 = [info[2]]
    box_coordiante2 = [info[3]]
    box_coordiante3 = [info[4]]
    box_coordiante4 = [info[5]]
    

    to:

    image_index = int(info[0]) // number: int(info[0])
    box_coordiante1 = info[2]
    box_coordiante2 = info[3]
    box_coordiante3 = info[4]
    box_coordiante4 = info[5]