Skip to content

boostcampaitech5/level3_cv_finalproject-cv-03

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

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

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