Vikramjeet Singh commited on
Commit
f55e094
1 Parent(s): 7d26070

Update README.md

Browse files

Former-commit-id: c301d8868c0c5ef68d1be144d4acd5baf893eaef

Files changed (1) hide show
  1. README.md +8 -3
README.md CHANGED
@@ -7,6 +7,10 @@
7
  - GPU used Nvidia A100 40GB
8
 
9
 
 
 
 
 
10
 
11
  ## Installation
12
 
@@ -25,14 +29,15 @@ This will install all necessary libraries for this project, including PIL , Diff
25
  python run.py --image_path /path/to/image.jpg --prompt 'prompt' --negative_prompt 'negative prompt' --output_dir /path/to/output --mask_dir /path/to/mask --uid unique_id
26
 
27
  ```
28
- ### Some Experiments
29
- Here are some of my experiments with the following models
30
  - https://huggingface.co/runwayml/stable-diffusion-inpainting
31
  - https://huggingface.co/lllyasviel/sd-controlnet-seg
32
  - https://huggingface.co/kandinsky-community/kandinsky-2-2-decoder-inpaint
33
 
 
34
 
35
-
36
 
37
  ![cooker_output](https://github.com/VikramxD/product_diffusion_api/assets/72499426/1228718b-5ef7-44a1-81f6-2953ffdc767c)
38
  ![toaster_output](https://github.com/VikramxD/product_diffusion_api/assets/72499426/06e12aea-cdc2-4ab8-97e0-be77bc49a238)
 
7
  - GPU used Nvidia A100 40GB
8
 
9
 
10
+ ##ARCHITECTURE
11
+
12
+ ![Architecture drawio](https://github.com/VikramxD/product_diffusion_api/assets/72499426/5a2e8b47-5a77-485b-b20c-0bca0928cb8a)
13
+
14
 
15
  ## Installation
16
 
 
29
  python run.py --image_path /path/to/image.jpg --prompt 'prompt' --negative_prompt 'negative prompt' --output_dir /path/to/output --mask_dir /path/to/mask --uid unique_id
30
 
31
  ```
32
+ ### MODELS USED
33
+ EXPERIMENTATION WITH THE FOLLOWING models
34
  - https://huggingface.co/runwayml/stable-diffusion-inpainting
35
  - https://huggingface.co/lllyasviel/sd-controlnet-seg
36
  - https://huggingface.co/kandinsky-community/kandinsky-2-2-decoder-inpaint
37
 
38
+ ### WEIGHTS AND BIASES EXPERIMENTATION REPORT
39
 
40
+ [](https://wandb.ai/vikramxd/product_placement_api/reports/Experimentation-Report---Vmlldzo3Mjg1MjQw)
41
 
42
  ![cooker_output](https://github.com/VikramxD/product_diffusion_api/assets/72499426/1228718b-5ef7-44a1-81f6-2953ffdc767c)
43
  ![toaster_output](https://github.com/VikramxD/product_diffusion_api/assets/72499426/06e12aea-cdc2-4ab8-97e0-be77bc49a238)