Increase JPEG output quality & disable subsampling (#801)

* Increase JPEG output quality & disable subsampling

Increased to JPEG95 from the default JPEG75 which is way too compressed. Output image size is now ~100kb. Previously was ~20kb.

* Increase JPEG output quality & disable subsampling

Add jpeg quality increase on cli

* line length changes

* line length changes

shark/examples/shark_inference/stable_diffusion/main.py

@@ -273,6 +273,8 @@ if __name__ == "__main__":
         }
         prompt_slice = re.sub("[^a-zA-Z0-9]", "_", args.prompts[i][:15])
         img_name = f"{prompt_slice}_{args.seed}_{i}_{dt.now().strftime('%y%m%d_%H%M%S')}"
-        pil_images[i].save(output_path / f"{img_name}.jpg")
+        pil_images[i].save(
+            output_path / f"{img_name}.jpg", quality=95, subsampling=0
+        )
         with open(output_path / f"{img_name}.json", "w") as f:
             f.write(json.dumps(json_store, indent=4))

shark/examples/shark_inference/stable_diffusion/sd_annotation.py

@@ -44,7 +44,10 @@ if args.use_winograd:
     download_public_file(full_gs_url, winograd_config_dir, True)
 
 if args.annotation_model == "unet" or device == "cuda":
-    if args.variant in ["anythingv3", "analogdiffusion"] or args.annotation_model == "vae":
+    if (
+        args.variant in ["anythingv3", "analogdiffusion"]
+        or args.annotation_model == "vae"
+    ):
         args.max_length = 77
     config_name = f"{args.annotation_model}_{args.version}_{args.precision}_len{args.max_length}_{device}.json"
     full_gs_url = config_bucket + config_name

web/models/stable_diffusion/main.py

@@ -81,7 +81,7 @@ def save_output_img(output_img):
         f"{prompt_slice}_{args.seed}_{dt.now().strftime('%y%m%d_%H%M%S')}"
     )
     out_img_path = Path(generated_imgs_path, f"{out_img_name}.jpg")
-    output_img.save(out_img_path)
+    output_img.save(out_img_path, quality=95, subsampling=0)
 
     new_entry = {
         "VARIANT": args.variant,