diff options
| author | Hiroshiba Kazuyuki <hihokaruta@gmail.com> | 2018-01-15 08:24:27 +0900 |
|---|---|---|
| committer | Hiroshiba Kazuyuki <hihokaruta@gmail.com> | 2018-01-15 08:24:27 +0900 |
| commit | c0f3eacabde5d41992a5ae1d8d8f0f170f6b155e (patch) | |
| tree | 83ca0a21583155ac19afd7ec2eab77b4293c2860 /become_yukarin/model/sr_model.py | |
| parent | 7bfc3321e356f24f49c790b578917e8db22bd30d (diff) | |
超解像クラス
Diffstat (limited to 'become_yukarin/model/sr_model.py')
| -rw-r--r-- | become_yukarin/model/sr_model.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/become_yukarin/model/sr_model.py b/become_yukarin/model/sr_model.py index 64158ca..8f9a34b 100644 --- a/become_yukarin/model/sr_model.py +++ b/become_yukarin/model/sr_model.py @@ -111,7 +111,11 @@ class SRDiscriminator(chainer.Chain): return h +def create_predictor_sr(config: SRModelConfig): + return SRPredictor(in_ch=1, out_ch=1) + + def create_sr(config: SRModelConfig): - predictor = SRPredictor(in_ch=1, out_ch=1) + predictor = create_predictor_sr() discriminator = SRDiscriminator(in_ch=1, out_ch=1) return predictor, discriminator |