AI_v2_pipeline_test_for_vision - 100-hours-a-week/16-Hot6-wiki GitHub Wiki
v2 pipeline
-
Load Models β Load Image β Grounding dino β (Make Prompt) β SDXL Inpainting β Image Upscaling β Labeling
-
μ΄λ² Testμμλ Make Prompt λΆλΆμ μλ΅νκ³ μλμ κ°μ΄ λ체
- recommend = ["potted plant", "ambient lamp", "LED light", "coffee mug", "desk clock"]
- prompt = "A cozy, add a stylish potted plant, an ambient lamp, soft LED lighting, a ceramic coffee mug, and a small modern desk clock to the existing desk setup."
-
pipeline colab νμΌμ λ°λ‘ 곡μ
Result
- {'ambient lamp light': [(929, 422)]}
- Labeling λμ§ μμ
- {'ambient lamp led light': [(438, 166)], 'desk clock': [(267, 289)]}
- Labeling λμ§ μμ
- {'ambient lamp led light': [(554, 343)]}
Issue
- Labelμ΄ μ λλ‘ λμ§ μλ λ¬Έμ
- Model Loadμ VRAMμ΄ μ½ 8.4GB μ¬μ©μ΄μ§λ§ 5μ₯μ λλ¦¬κ³ λ ν VRAMμ΄ μ½ 8.5GBμΌλ‘ μν μμΉ, μ΄λμ κ° λ©λͺ¨λ¦¬ λμμ€