Go to file
2023-02-01 22:14:43 +01:00
DCT-Net@cf416a632f Add style transfer submodule 2023-01-15 11:52:04 +01:00
haarcascades More robust face detection 2023-02-01 20:25:31 +01:00
test_set create test set 2023-01-29 19:02:39 +01:00
.editorconfig Added method outline 2023-01-15 12:40:35 +01:00
.gitignore Yolo face detection 2023-02-01 21:09:36 +01:00
.gitmodules Add style transfer submodule 2023-01-15 11:52:04 +01:00
face_detect.py Yolo face detection 2023-02-01 21:09:36 +01:00
helpers.py Refactor and bug fixes 2023-02-01 19:55:12 +01:00
load_test_data.py Refactor and bug fixes 2023-02-01 19:55:12 +01:00
main.py Use correlation 2023-02-01 21:27:39 +01:00
metrics.py Refactor and bug fixes 2023-02-01 19:55:12 +01:00
plots.py Plot results 2023-02-01 13:16:46 +01:00
README.md Add README.md 2023-02-01 12:28:51 +01:00
requirements-osx.txt Add README.md 2023-02-01 12:28:51 +01:00
requirements.txt Yolo face detection 2023-02-01 21:09:36 +01:00
scrape_data.py scrap n images depends on argv param 2023-01-15 14:17:33 +01:00
UAM-Andre.jpg find and crop face implementation 2023-01-29 15:17:39 +01:00

wko_anime-face-similarity

Projekt przygotowany na zajęcia z widzenia komputerowego.

Rozpoznaje twarz na zdjęciu wejściowym i dokonując transferu stylu do anime, porównuje zdjęcie ze zbiorem postaci z anime i wskazuje podobieństwa według wybranych metryk.

Instalacja

  1. Pobranie submodułów:
    $ git submodule update --init
    
  2. Instalacja zależności:
    • Windows/Linux
      $ pip install -r requirements.txt     
      
    • MacOS
      $ pip install -r requirements-osx.txt
      
  3. Konfiguracja DCT-Netu (anime style transfer)
    $ cd DCT-Net && python download.py
    
  4. Pobranie datasetu twarzy postaci z anime (MyAnimeList)
    $ python scrape_data.py
    

Uruchomienie

Na tę chwilę zdjęcie poddawane porównaniu to UAM-Andre.jpg

$ python main.py

Walidacja

Do walidacji metryk na postawie testowego datasetu z cosplayerami (test_set) uruchamiamy

$ python --validate_only 1