{"id":24378227,"url":"https://github.com/wiktorlazarski/edge-aware-face-autoencoder","last_synced_at":"2026-04-25T19:31:56.899Z","repository":{"id":38311086,"uuid":"475504999","full_name":"wiktorlazarski/edge-aware-face-autoencoder","owner":"wiktorlazarski","description":"👨‍🎨 Edge-aware Face Autoencoder.","archived":false,"fork":false,"pushed_at":"2022-06-07T15:58:59.000Z","size":97668,"stargazers_count":1,"open_issues_count":0,"forks_count":1,"subscribers_count":3,"default_branch":"master","last_synced_at":"2025-01-19T06:19:59.127Z","etag":null,"topics":["autoencoder","celeba-dataset","computer-vision","deep-learning","face","generative-model","hydra","image-compression","pytorch","pytorch-lightning","vae-pytorch","wandb"],"latest_commit_sha":null,"homepage":"https://colab.research.google.com/drive/1b3afUvKmuWQblkxIpGxUAOt0pZk3DBa3?usp=sharing","language":"Jupyter Notebook","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"other","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/wiktorlazarski.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null}},"created_at":"2022-03-29T15:28:30.000Z","updated_at":"2022-06-07T17:18:42.000Z","dependencies_parsed_at":"2022-08-25T21:22:15.710Z","dependency_job_id":null,"html_url":"https://github.com/wiktorlazarski/edge-aware-face-autoencoder","commit_stats":null,"previous_names":[],"tags_count":2,"template":false,"template_full_name":"wiktorlazarski/ai-awesome-project-template","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/wiktorlazarski%2Fedge-aware-face-autoencoder","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/wiktorlazarski%2Fedge-aware-face-autoencoder/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/wiktorlazarski%2Fedge-aware-face-autoencoder/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/wiktorlazarski%2Fedge-aware-face-autoencoder/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/wiktorlazarski","download_url":"https://codeload.github.com/wiktorlazarski/edge-aware-face-autoencoder/tar.gz/refs/heads/master","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":243222183,"owners_count":20256220,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["autoencoder","celeba-dataset","computer-vision","deep-learning","face","generative-model","hydra","image-compression","pytorch","pytorch-lightning","vae-pytorch","wandb"],"created_at":"2025-01-19T06:18:53.058Z","updated_at":"2025-12-11T19:17:08.310Z","avatar_url":"https://github.com/wiktorlazarski.png","language":"Jupyter Notebook","funding_links":[],"categories":[],"sub_categories":[],"readme":"______________________________________________________________________\n\u003cdiv align=\"center\"\u003e\n\n# 👨‍🎨 Edge-aware Face Autoencoder\n\n\u003cp align=\"center\"\u003e\n  \u003ca href=\"https://github.com/wiktorlazarski\"\u003e👨‍🎓 Wiktor\u003c/a\u003e\n  \u003ca href=\"https://github.com/AnetaJas\"\u003e👩‍🎓 Aneta\u003c/a\u003e\n\u003c/p\u003e\n\n______________________________________________________________________\n\n[![ci-testing](https://github.com/wiktorlazarski/face-vae/actions/workflows/ci-testing.yml/badge.svg?branch=master\u0026event=push)](https://github.com/wiktorlazarski/edge-aware-face-autoencoder/actions/workflows/ci-testing.yml)\n[![Open In Collab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1b3afUvKmuWQblkxIpGxUAOt0pZk3DBa3?usp=sharing)\n[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)\n\n\u003c/div\u003e\n\n## 💎 Installation with `pip`\n\nInstallation is as simple as running:\n\n```bash\npip install git+https://github.com/wiktorlazarski/edge-aware-face-autoencoder.git\n```\n\n## 🧐 Qualitative results\n\n#### Original images\n![alt text](https://github.com/wiktorlazarski/edge-aware-face-autoencoder/blob/master/doc/images/original_imgs.png)\n\n---\n\n#### Baseline model without edge-awareness trained on images with 256px resolution, VAE latent dimension equals to 512 and reconstruction loss weight equals to 100 000. Edges' weight was set to 1.\n![alt text](https://github.com/wiktorlazarski/edge-aware-face-autoencoder/blob/master/doc/images/baseline-256px-512ld-100k.png)\n\n---\n\n#### Edge-aware model with edges' weight set to 3. All other parameters were the same as in the baseline model.\n![alt text](https://github.com/wiktorlazarski/edge-aware-face-autoencoder/blob/master/doc/images/edgeaware-edge_weight3.png)\n\n---\n\n#### Edge-aware model with edges' weight set to 10. All other parameters were the same as in the baseline model.\n![alt text](https://github.com/wiktorlazarski/edge-aware-face-autoencoder/blob/master/doc/images/edgeaware_edge_weight10.png)\n\n---\n\n## ⚙️ Setup for development with `pip`\n\n```bash\n# Clone repo\ngit clone https://github.com/wiktorlazarski/edge-aware-face-autoencoder.git\n\n# Go to repo directory\ncd edge-aware-face-autoencoder\n\n# (Optional) Create virtual environment\npython -m venv venv\nsource ./venv/bin/activate\n\n# Install project in editable mode\npip install -e .[dev]\n\n# (Optional but recommended) Install pre-commit hooks to preserve code format consistency\npre-commit install\n```\n\n## 🐍 Setup for development with `conda`\n\n```bash\n# Clone repo\ngit clone https://github.com/wiktorlazarski/edge-aware-face-autoencoder.git\n\n# Go to repo directory\ncd edge-aware-face-autoencoder\n\n# Create and activate conda environment\nconda env create -f ./conda_env.yml\nconda activate face_autoencoder\n\n# (Optional but recommended) Install pre-commit hooks to preserve code format consistency\npre-commit install\n```\n\n\u003cdiv align=\"center\"\u003e\n\n### 🤗 Enjoy the model !\n\n\u003c/div\u003e\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fwiktorlazarski%2Fedge-aware-face-autoencoder","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fwiktorlazarski%2Fedge-aware-face-autoencoder","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fwiktorlazarski%2Fedge-aware-face-autoencoder/lists"}