Updated for V2.0

model_inferencing.py

@@ -1,8 +1,6 @@
 TargetModel = None
 def generate_test_image(T2IModel, testPrompt):
- #prompt = "The quick brown fox jumps over the lazy dog"
  testImage = TargetModel(testPrompt, num_inference_steps=50).images[0]
- #testImage.save("./image.png")
  
  return testImage
 
@@ -22,13 +20,13 @@ def construct_general_bias_evaluation_prompts(subjects, actions):
  prompts.append('a picture of a ' + subject)
 
  return prompts
-def generate_test_images(progressBar, barText, prompts, NSamples, NSteps, imageSize):
+def generate_test_images(progressBar, barText, prompts, NSamples, NSteps, imageWidth, imageHeight):
  guidance = 7.5
  testImages = []
  imageCaptions = [[], []]
  for prompt, ii in zip(prompts, range(len(prompts))):
  testImages+=TargetModel(prompt, num_images_per_prompt=NSamples, num_inference_steps=NSteps,
- guidance_scale=guidance, width=imageSize, height=imageSize).images
+ guidance_scale=guidance, width=imageWidth, height=imageHeight).images
  for nn in range(NSamples):
  imageCaptions[0].append(prompt) # actual prompt used
  imageCaptions[1].append("Prompt: "+str(ii+1)+" Sample: "+ str(nn+1)) # caption for the image output
@@ -38,13 +36,13 @@ def generate_test_images(progressBar, barText, prompts, NSamples, NSteps, imageS
  progressBar.empty()
  return (testImages, imageCaptions)
 
-def generate_task_oriented_images(progressBar, barText, prompts, ids, NSamples, NSteps, imageSize):
+def generate_task_oriented_images(progressBar, barText, prompts, ids, NSamples, NSteps, imageWidth, imageHeight):
  guidance = 7.5
  testImages = []
  imageCaptions = [[], []]
  for prompt, jj in zip(prompts, range(len(prompts))):
  testImages+=TargetModel(prompt, num_images_per_prompt=NSamples, num_inference_steps=NSteps,
- guidance_scale=guidance, width=imageSize, height=imageSize).images
+ guidance_scale=guidance, width=imageWidth, height=imageHeight).images
  for nn in range(NSamples):
  imageCaptions[0].append(prompt) # actual prompt used
  imageCaptions[1].append("COCO ID: "+ids[jj]+" Sample: "+ str(nn+1)) # caption for the image output