TencentARC/GFPGAN
1
0
Fork 0
mirror of https://github.com/TencentARC/GFPGAN.git synced 2026-04-19 04:10:47 +00:00
Code Issues 476 Packages Projects Releases 8 Wiki Activity

Training on custom dataset #36

New Issue
Open
opened 2026-01-29 21:38:56 +00:00 by claunia · 0 comments
claunia commented 2026-01-29 21:38:56 +00:00
Owner
Copy Link

Originally created by @3BBUAE on GitHub (Jul 28, 2021).

Hello, first of all fantastic work and thanks for sharing.

I would like to know how can I train the model on a custom dataset?
I noticed in the training explanation, there are 3 files I need to download excluding the dataset:
-Pre trained styleGAN2 model
-FFHQ component locations
-Arcface

I know that Arcface is used for face recognition. I assume that the pretrained styleGAN2 model, is to train the GFPGAN model from scratch so if i wanted to continue training I could just use the model you have provided for inference to continue training on my custom dataset. Finally for the components locations file, will it work with my custom dataset or is it specific to the FFHQ dataset, and if it won't work how will I be able to create my own so it may work with my dataset?

I hope my issue is clear, thanks.

Originally created by @3BBUAE on GitHub (Jul 28, 2021). Hello, first of all fantastic work and thanks for sharing. I would like to know how can I train the model on a custom dataset? I noticed in the training explanation, there are 3 files I need to download excluding the dataset: -Pre trained styleGAN2 model -FFHQ component locations -Arcface I know that Arcface is used for face recognition. I assume that the pretrained styleGAN2 model, is to train the GFPGAN model from scratch so if i wanted to continue training I could just use the [model ](https://github.com/TencentARC/GFPGAN/releases/download/v0.1.0/GFPGANv1.pth) you have provided for inference to continue training on my custom dataset. Finally for the components locations file, will it work with my custom dataset or is it specific to the FFHQ dataset, and if it won't work how will I be able to create my own so it may work with my dataset? I hope my issue is clear, thanks.
Sign in to join this conversation.
No Branch/Tag Specified
Branches Tags
Labels
Clear labels
pull-request
Mirrored from GitHub Pull Request
No Label
Milestone
No items
No Milestone
Projects
Clear projects
No project
Assignees
Clear assignees
claunia
No Assignees
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: TencentARC/GFPGAN#36
Delete Branch "%!s()"

Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?