Está sorteando os objetos corretamente agora

aumento_dados_coco.py

@@ -76,6 +76,9 @@ licenses = [{
     "name": "Attribution License",
 }]
 
+# Pega a lista de objetos disponíveis
+objetos = os.listdir(OBJETOS_DIR)
+
 for fundo in os.listdir(FUNDOS_DIR):
     print("Gerando variações para o fundo ", fundo)
 
@@ -83,11 +86,6 @@ for fundo in os.listdir(FUNDOS_DIR):
 
         i_img = len(coco.imgs)+1
         print("Criando imagem " + str(i_img))
-        # Seleciona uma imagem aleatória do diretórios de objetos
-        objetos = os.listdir(OBJETOS_DIR)
-        objeto = random.choice(objetos)
-        # Coloca em IMG_PATH o caminho completo da imagem do objeto
-        IMG_PATH = OBJETOS_DIR + "/" + objeto
 
         # Carrega a imagem de fundo de dentro da pasta FUNDOS_DIR usando o OpenCV
         img = cv2.imread(FUNDOS_DIR + "/" + fundo)
@@ -125,6 +123,10 @@ for fundo in os.listdir(FUNDOS_DIR):
 
             # Imprime j seguido de '...' sem pular de linha
             print(j+1, end="...")
+            
+            objeto = random.choice(objetos)
+            # Coloca em IMG_PATH o caminho completo da imagem do objeto
+            IMG_PATH = OBJETOS_DIR + "/" + objeto
 
             # Cria variações do objeto
             response = openai.Image.create_variation(