Skip to content

boostcampaitech5/level3_cv_finalproject-cv-03

Folders and files

NameName
Last commit message
Last commit date

Latest commit

1184763 Β· Dec 12, 2023
Jul 27, 2023
Aug 1, 2023
Aug 9, 2023
Aug 2, 2023
Jul 27, 2023
Jul 27, 2023
Jul 26, 2023
Aug 2, 2023
Dec 12, 2023
Jul 5, 2023
Jul 30, 2023
Jul 27, 2023

Repository files navigation


πŸ’½ 앨범 ν‘œμ§€ 맞좀 μ œμž‘ μ„œλΉ„μŠ€ πŸ’½

✨ νŒ€ μ†Œκ°œ

Thanks goes to these wonderful people (emoji key):

This project follows the all-contributors specification. Contributions of any kind welcome!

πŸ’€ ν”„λ‘œμ νŠΈ μ†Œκ°œ

앨범 ν‘œμ§€λŠ” κ°€μˆ˜μ™€ 그의 νŒ¬λ“€ μ‚¬μ΄μ˜ 첫 λ§Œλ‚¨μž…λ‹ˆλ‹€. 처음 λ§Œλ‚˜λŠ” μˆœκ°„μ˜ μ€‘μš”μ„±μ€ μ²«μ‚¬λž‘, 첫 λ§Œλ‚¨, 첫 인상과 같은 κΈ°μ–΅ μ†μ—μ„œλ„ λŠκ»΄μ§‘λ‹ˆλ‹€. μ΄λŸ¬ν•œ μ€‘μš”μ„±μ€ μ˜ν™” ν¬μŠ€ν„°λ‚˜ 유튜브 썸넀일과 같이 앨범 컀버도 μž‘ν’ˆμ˜ λΆ„μœ„κΈ°μ™€ μ•„ν‹°μŠ€νŠΈμ˜ 이미지λ₯Ό λŒ€λ³€ν•©λ‹ˆλ‹€. ν•˜μ§€λ§Œ 앨범 μ œμž‘μ€ λ§Žμ€ λΉ„μš©κ³Ό μ‹œκ°„μ΄ μ†Œμš”λ˜λŠ” μž‘μ—…μž…λ‹ˆλ‹€. λ³Έ ν”„λ‘œμ νŠΈλŠ” μ΄λŸ¬ν•œ 과정을 λ‹¨μˆœν™”ν•˜μ—¬ λΉ λ₯΄κ³  μ‰½κ²Œ λ§žμΆ€ν˜• 앨범 컀버λ₯Ό μ œμž‘ν•˜λŠ” μ„œλΉ„μŠ€λ₯Ό μ œκ³΅ν•©λ‹ˆλ‹€.

ν•΄λ‹Ή ν”„λ‘œμ νŠΈλŠ” Boostcamp AI Tech λ‚΄μ—μ„œ μ§„ν–‰λœ μ΅œμ’… ν”„λ‘œμ νŠΈμž…λ‹ˆλ‹€.

πŸ“† ν”„λ‘œμ νŠΈ 일정

ν”„λ‘œμ νŠΈ 전체 일정

  • 2023.07.03 ~ 2023.07.28

ν”„λ‘œμ νŠΈ μ„ΈλΆ€ 일정

  • ~ 2023.06.25 : 주제 μ„ μ •
  • 2023.06.26 ~ 2023.06.30 : Dataset Search 및 Model Research
  • 2023.07.01 ~ 2023.07.02 : Dataset Crawling
  • 2023.07.03 ~ 2023.07.11 : Stable Diffusion ν•™μŠ΅ νŒŒμ΄ν”„λΌμΈ ꡬ좕 및 FastAPIλ₯Ό μ΄μš©ν•œ Scratch μ œμž‘
  • 2023.07.11 ~ 2023.07.12 : BigQuery와 Google Cloud Storageλ₯Ό μ‚¬μš©ν•΄ DataWarehouse ꡬ좕
  • 2023.07.12 ~ 2023.07.13 : Error Reporting
  • 2023.07.13 ~ 2023.07.28 : Dreambooth와 Stable Diffusion을 ν™œμš©ν•œ Fine-Tuning μ‹€ν—˜
  • 2023.07.12 ~ 2023.07.28 : Celery, Redis, FastAPIλ₯Ό ν™œμš©ν•œ Back-End ꡬ좕
  • 2023.07.17 ~ 2023.07.28 : Front-End와 Github Action을 ν™œμš©ν•œ CI ꡬ좕, Airflow μ—°κ²°
  • 2023.07.28 ~ : ν”„λ‘œμ νŠΈ 마무리 및 배포

🚧 ν”„λ‘œμ νŠΈ ꡬ쑰

πŸ‘¨β€πŸ’» ν”„λ‘œμ νŠΈ μˆ˜ν–‰

  1. Datasetβœ”οΈ
  2. Stable Diffusionβœ”οΈ
  3. Dreamboothβœ”οΈ
  4. Scratchβœ”οΈ
  5. Experimentsβœ”οΈ
  6. λ°œν‘œμžλ£Œβ­
  7. λ°œν‘œμ˜μƒβ­

πŸ—’οΈ ν”„λ‘œμ νŠΈ κ²°κ³Ό

πŸ”ΈDataset

  • 데이터 μˆ˜μ§‘μ΄ νŽΈλ¦¬ν•˜κ³  κ΅­λ‚΄/κ΅­μ™Έ λ…Έλž˜ 데이터 골고루 μˆ˜μ§‘κ°€λŠ₯ν•œ 멜둠 μŒμ›μ‚¬μ΄νŠΈ 선택
  • ν•΄λ‹Ή μŒμ›μ‚¬μ΄νŠΈλ‘œλΆ€ν„° λ…Έλž˜ 제λͺ©, κ°€μˆ˜ 이름, 앨범 이 름, 발맀일, λ…Έλž˜ μž₯λ₯΄, λ…Έλž˜ 가사, 앨범 ν‘œμ§€ URL을 Crawlingν•˜μ—¬ 총 3,851개의 데이터 μˆ˜μ§‘

πŸ”ΈModel

1. Stable Diffusion

Stable Diffusion은 높은 ν’ˆμ§ˆμ˜ 이미지λ₯Ό μƒμ„±ν•˜κ³ , Text-To-Image μž‘μ—…μ„ κ°€λŠ₯ν•˜κ²Œ ν•΄μ£ΌλŠ” λͺ¨λΈμž…λ‹ˆλ‹€. μ‚¬μš©μžλ‘œλΆ€ν„° μž…λ ₯ 받은 λ…Έλž˜ 정보λ₯Ό ChatGPT APIλ₯Ό 톡해 Prompt둜 μƒμ„±ν•œ ν›„, 이λ₯Ό Stable Diffusion λͺ¨λΈμ— μž…λ ₯ν•˜μ—¬ 앨범 컀버λ₯Ό μƒμ„±ν•©λ‹ˆλ‹€.

2. Dreambooth

DreamboothλŠ” Stable Diffusionκ³ΌλŠ” 달리, μ‚¬μš©μžλ‘œλΆ€ν„° μΆ”κ°€μ μœΌλ‘œ κ°€μˆ˜μ˜ 사진을 μž…λ ₯ λ°›κ³  이λ₯Ό 톡해 Fine-Tuningν•˜μ—¬ κ°€μˆ˜ 사진이 ν¬ν•¨λœ 앨범 컀버λ₯Ό μƒμ„±ν•΄μ€λ‹ˆλ‹€.


πŸ”„οΈ Directory

β”œβ”€β”€ .gitignore
β”œβ”€β”€ .pre-commit-config.yaml
β”œβ”€β”€ gitcommit_template.txt
β”œβ”€β”€ README.md
β”œβ”€β”€ Makefile
β”œβ”€β”€ poetry.lock
β”œβ”€β”€ pyproject.toml
β”œβ”€β”€ imgs
β”œβ”€β”€ docs
β”œβ”€β”€ tests
└── src
    β”œβ”€β”€  scratch
    β”‚    β”œβ”€β”€ apache_config
    β”‚    β”œβ”€β”€ config
    β”‚    β”œβ”€β”€ dags
    β”‚    β”‚    β”œβ”€β”€ bigquery_data_pipeline.py
    β”‚    β”‚    └── model_retrain.py
    β”‚    β”œβ”€β”€ dreambooth
    β”‚    β”‚    β”œβ”€β”€ dreambooth.yaml
    β”‚    β”‚    β”œβ”€β”€ inference.py
    β”‚    β”‚    └── run.py
    β”‚    β”œβ”€β”€ gcp
    β”‚    β”‚    β”œβ”€β”€ bigquery.py
    β”‚    β”‚    β”œβ”€β”€ cloud_storage.py
    β”‚    β”‚    └── error.py
    β”‚    β”œβ”€β”€ htdocs
    β”‚    β”‚    β”œβ”€β”€ images
    β”‚    β”‚    β”œβ”€β”€ js
    β”‚    β”‚    β”œβ”€β”€ styles
    β”‚    β”‚    β”œβ”€β”€ create.html
    β”‚    β”‚    β”œβ”€β”€ index.html
    β”‚    β”‚    └── mypage.html
    β”‚    β”œβ”€β”€ gpt3-api.py
    β”‚    β”œβ”€β”€ main.py
    β”‚    β”œβ”€β”€ model.py
    β”‚    β”œβ”€β”€ streamlit_frontend.py
    β”‚    β”œβ”€β”€ train_dreambooth.py
    β”‚    β”œβ”€β”€ translation.py
    β”‚    β”œβ”€β”€ utils.py
    β”‚    β”œβ”€β”€ worker_dreambooth.py
    β”‚    β”œβ”€β”€ worker_sd.py
    β”‚    └── worker_sdxl.py
    β”‚
    β”œβ”€β”€ stable_diffusion
    β”‚    β”œβ”€β”€ dataset.py
    β”‚    β”œβ”€β”€ main.py
    β”‚    β”œβ”€β”€ test.py
    β”‚    β”œβ”€β”€ train.py
    β”‚    β”œβ”€β”€ utils
    β”‚    β”‚    β”œβ”€β”€ plot.py
    β”‚    β”‚    β”œβ”€β”€ training.py
    β”‚    β”‚    └── util.py
    β”‚    └── experiments
    β”‚         └── [tag]ExpName
    β”‚			    β”œβ”€β”€ albums.csv
    β”‚			    β”œβ”€β”€ prompts.txt
    β”‚			    β”œβ”€β”€ checkpoints
    β”‚			    └── results
    β”‚			     
    └── stable_diffusion_xl 
         └── (stable_diffusionκ³Ό ꡬ쑰 동일)

πŸ€” Wrap-Up Report

βš οΈμ €μž‘κΆŒ ν‘œκΈ°