Skip to content
This repository has been archived by the owner on Oct 11, 2023. It is now read-only.

Pretrained weights #11

Closed
muaz-urwa opened this issue Nov 5, 2019 · 5 comments
Closed

Pretrained weights #11

muaz-urwa opened this issue Nov 5, 2019 · 5 comments
Assignees

Comments

@muaz-urwa
Copy link

Thank you for the awesome project. It would be awesome if you can also share the pretrained weights (checkpoints) with it.
Would it be possible ?

@andrewjong
Copy link
Owner

andrewjong commented Nov 6, 2019

I need to retrain them because I was simply prototyping SwapNet before, and deleted them by mistake. But note the implementation is only 90% finished. There are still two things to implement/fix, namely (1) Copy the face data in the final result (issue #6), and (2) debug the perceptual loss to achieve high texture quality (issue #7). A released model at this time would not have those included. Would you like a model with this status or do you prefer to wait for a final product?

I'm also currently busy until mid-December (working on grad school applications). The earliest I can promise is late December.

@muaz-urwa
Copy link
Author

Dear Andrew,
I would really appreciate it if you can share what you have right now.

Good luck for the grad school application !

@andrewjong andrewjong self-assigned this Dec 18, 2019
@andrewjong andrewjong added this to the Sprint December milestone Dec 19, 2019
@andrewjong andrewjong removed this from the December Sprint milestone Jan 9, 2020
@Larkinnjm1
Copy link

Hey folks, thanks once again for uploading this unbelievable repo. Did you get any joy uploading your training weights?

@andrewjong
Copy link
Owner

Hi all, thank you for your patience! And thank you @Larkinnjm1 for the reminder. I just started training a model for upload. Going to let it run overnight, then upload tomorrow evening.

Proof:
Screenshot_20200113_2146001

@andrewjong
Copy link
Owner

andrewjong commented Jan 15, 2020

Update: models have finished training. Download here and extract under the project root: https://drive.google.com/open?id=1kaeHlGf9h3vZLBxr4D5yQUgToxEUWuMh

Please note that these models are far from perfect. I have yet to properly implement the Perceptual Loss, which is crucial for proper texture transfer. See #7 .

Let me know if there are any issues!

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants