Reputation: 45
I am trying to display rectangles in an image using cv2
. However due to the following error I cannot make them on the image.
I have converted the image into numpy
array and tried to write down the vertices in different ways but the general format is correct. I don't know from where a tuple error is showing up.
def draw_rect(img, dims, color = None):
img = img.copy()
dims = dims.reshape(-1, 4)
if not color:
color = (255, 255, 255) #rgb values for white color
for dim in dims:
x, y = (dim[0], dim[1]) , (dim[2], dim[3])
x = int(x[0]), int(x[1])
y = int(y[0]), int(y[1])
img = cv2.rectangle(img, x, y, color, int(max(img.shape[:,2])/200)) # error
return img
def main():
addr = 'test1_sec0.jpg'
bbox = 'test1_sec0.jpg.csv'
img_show(addr) # used to read and show the image using cv2. Works fine.
dims = bbox_read(bbox) # used to read the boundary boxes. Works fine
img = cv2.imread(addr, 1)
img_data = np.asarray(img, dtype = 'int32')
print(img_data)
plt.imshow(draw_rect(img_data, dims)) # error
plt.show( )
if __name__ == '__main__':
main()
'
Traceback (most recent call last):
File "ImgAug.py", line 56, in <module>
main()
File "ImgAug.py", line 51, in main
plt.imshow(draw_rect(img_data, dims))
File "ImgAug.py", line 41, in draw_rect
img = cv2.rectangle(img, x[0], x[1], y[0], y[1], color, int(max(img.shape[:,2])/200)) # first argument & variable must be same cause same image should have all the bbox
TypeError: tuple indices must be integers or slices, not tuple
Upvotes: 1
Views: 2070
Reputation: 44926
img.shape
is a tuple, but img.shape[:,2]
is trying to index it with another tuple, which is invalid:
>>> class X:
... def __getitem__(self, index):
... print(f'The index is: {index}')
...
>>> X()[:,2]
The index is: (slice(None, None, None), 2)
As you can see, something[:,2]
actually generates a tuple as the index.
Upvotes: 2