Я пытаюсь сгенерировать набор данных CLEVR с некоторыми случайными изображениями прямоугольников и окружностей. Когда я отлаживаю код для каждого оператора plt.imshow(), я вижу новое изображение с формой, сгенерированной вместе с формами, сгенерированными в последних выполненных циклах. Хотя, к моему удивлению, каждый раз 5-е изображение не показывается в режиме отладки.
Также, когда я сохраняю изображения локально, я вижу только два объекта. Пожалуйста, найдите изображения для справочных целей.
import matplotlib.pyplot as plt
import cv2
import numpy as np
colors = [
(0, 0, 255), # r
(0, 255, 0), # g
(255, 0, 0), # b
(0, 156, 255), # o
(128, 128, 128), # k
(0, 255, 255) # y
]
img_size = 75
size = 5
def center_generate(objects):
while True:
pas = True
center = np.random.randint(0+size, img_size - size, 2)
if len(objects) > 0:
for name, c, shape in objects:
"""this condition checks if the new object generated is not completely covered by the already
generated objects"""
if ((center - c) ** 2).sum() < ((size * 2) ** 2):
pas = False
if pas:
return center
def build_dataset(dataset_type, loopnumber):
objects = []
img = np.ones((img_size, img_size, 3)) * 255
for color_id, color in enumerate(colors):
center = center_generate(objects)
if random.random() < 0.5:
start = (center[0]-size, center[1]-size)
end = (center[0]+size, center[1]+size)
img = cv2.rectangle(img, start, end, color, -1)
plt.imshow(img)
objects.append((color_id, center, 'r'))
else:
center_ = (center[0], center[1])
img = cv2.circle(img, center_, size, color, -1)
plt.imshow(img)
objects.append((color_id, center, 'c'))
if dataset_type == 'test':
plt.imsave('..\\img\\test\\img_' + str(loop) + '.jpeg', img)
else:
plt.imsave('..\\img\\train\\img_' + str(loop) + '.jpeg', img)
Изображение, сгенерированное во время отладки.
Изображение хранится локально.
Может ли кто-нибудь объяснить эту проблему здесь?
Спасибо