Оптимизация производительности в DeblurGANv2Python

Программы на Python
Anonymous
Оптимизация производительности в DeblurGANv2

Сообщение Anonymous »

Код: Выделить всё

---
project: deblur_gan
experiment_desc: 11_50_fpn_stop

train:
#  files_a: &FILES_A /datasets/my_dataset/**/*.jpg
#可以使用绝对路径
#files_a: &FILES_A D:/deblur/goprol_large/**/*.png
#files_a: &FILES_A  C:/Users/user/PycharmProjects/DeblurGANv2-master/test_img/GOPRO/GOPRO_3840FPS_AVG_3-21/train/blur/**/*.png
files_a: &FILES_A D:/deblur/blur_11/*.jpg
files_b: &FILES_B D:/deblur/img_align_celeba/*.jpg
#  files_b: &FILES_B ./dataset1/sharp/*.png
size: &SIZE 256
crop: random
preload: &PRELOAD false
preload_size: &PRELOAD_SIZE 0
bounds: [0, .8]
scope: geometric
corrupt: &CORRUPT
- name: cutout
prob: 0.5
num_holes: 3
max_h_size: 25
max_w_size: 25
- name: jpeg
quality_lower: 70
quality_upper: 90
- name: motion_blur
- name: median_blur
- name: gamma
- name: rgb_shift
- name: hsv_shift
- name: sharpen

val:
files_a: *FILES_A
files_b: *FILES_B
#  files_a: &FILES_A
#  files_b: &FILES_B
size: *SIZE
scope: geometric
crop: center
preload: *PRELOAD
preload_size: *PRELOAD_SIZE
bounds: [.8, 1]
corrupt: *CORRUPT

phase: train
warmup_num: 3
model:
g_name: fpn_inception
blocks: 9
d_name: double_gan # may be no_gan, patch_gan, double_gan, multi_scale
d_layers: 3
content_loss: perceptual
adv_lambda: 0.001
disc_loss: ragan-ls
learn_residual: True
norm_layer: instance
dropout: True

num_epochs: 50
train_batches_per_epoch: 1000
val_batches_per_epoch: 100
batch_size: 1
image_size: [178, 218]

optimizer:
name: adam
lr: 0.0001
scheduler:
name: linear
start_epoch: 50
min_lr: 0.0000001
В настоящее время я использую DeblurGANv2 с набором данных из 6000 изображений и обучаюсь с размером пакета 1. Я проводил эксперименты с 10, 30 и 50 эпохами. После восстановления изображений я оценил производительность с помощью показателей PSNR, SSIM и FID. Однако я заметил следующее:
По мере увеличения количества эпох производительность не улучшается.
В некоторых случаях показатели производительности фактически снижаются.
Не могли бы вы подсказать, какие параметры мне следует настроить, чтобы добиться большей производительности в большем количестве эпох?

Подробнее здесь: https://stackoverflow.com/questions/790 ... eblurganv2

Вернуться в «Python»