Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to fine-tune existing model? #24

Closed
AlexanderKozhevin opened this issue Feb 2, 2024 · 3 comments
Closed

How to fine-tune existing model? #24

AlexanderKozhevin opened this issue Feb 2, 2024 · 3 comments

Comments

@AlexanderKozhevin
Copy link

No description provided.

@piddnad
Copy link
Owner

piddnad commented Mar 15, 2024

Hi, you can refer to #3 (comment) and set pretrain_network_g as an existing model in the configuration file.

@mycomedico
Copy link

Do i need both the black and white and color images? or does the training automatically generate black and white images? do i need to specify seperate folders for the images if i need to include both? my images have resolution of 720height x 992 width. is this the correct format to specify that: gt_size: [720, 992]

@piddnad
Copy link
Owner

piddnad commented Apr 2, 2024

Do i need both the black and white and color images? or does the training automatically generate black and white images? do i need to specify seperate folders for the images if i need to include both? my images have resolution of 720height x 992 width. is this the correct format to specify that: gt_size: [720, 992]

  1. Only color images are needed. During training, color images will automatically converted to black and white images. So just set your color image path list file in "meta_info_file" (e.g. 'data_list/imagenet.txt').
  2. No, there is no need to change "gt_size". "gt_size" represents the resolution of GT samples and 256 is usually sufficient. When training, dataloader will automatically resize the input to the this size (see here) .

Hope this helps!

@piddnad piddnad closed this as completed May 3, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants