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.

Sign up for GitHub

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

Jump to bottom

[Fix] fix srgan train config #1441

Merged
merged 4 commits into from
Nov 16, 2022
Merged

[Fix] fix srgan train config #1441

merged 4 commits into from
Nov 16, 2022

Conversation

Z-Fran
Copy link
Collaborator

@Z-Fran Z-Fran commented Nov 16, 2022

Thanks for your contribution and we appreciate it a lot. The following instructions would make your pull request more healthy and more easily get feedback. If you do not understand some items, don't worry, just make the pull request and seek help from maintainers.

Motivation

Please describe the motivation of this PR and the goal you want to achieve through this PR.

Modification

Please briefly describe what modification is made in this PR.

Who can help? @ them here!

BC-breaking (Optional)

Does the modification introduce changes that break the backward-compatibility of the downstream repositories?
If so, please describe how it breaks the compatibility and how the downstream projects should modify their code to keep compatibility with this PR.

Use cases (Optional)

If this PR introduces a new feature, it is better to list some use cases here, and update the documentation.

Checklist

Before PR:

  • I have read and followed the workflow indicated in the CONTRIBUTING.md to create this PR.
  • Pre-commit or linting tools indicated in CONTRIBUTING.md are used to fix the potential lint issues.
  • Bug fixes are covered by unit tests, the case that causes the bug should be added in the unit tests.
  • New functionalities are covered by complete unit tests. If not, please add more unit test to ensure the correctness.
  • The documentation has been modified accordingly, including docstring or example tutorials.

After PR:

  • If the modification has potential influence on downstream or other related projects, this PR should be tested with some of those projects.
  • CLA has been signed and all committers have signed the CLA in this PR.

@Z-Fran Z-Fran requested a review from zengyh1900 November 16, 2022 03:50
@Z-Fran Z-Fran self-assigned this Nov 16, 2022
@zengyh1900 zengyh1900 merged commit 63ee765 into dev-1.x Nov 16, 2022
@zengyh1900 zengyh1900 deleted the fix-srgan branch November 16, 2022 09:12
zengyh1900 added a commit that referenced this pull request Dec 13, 2022
@zengyh1900 @Yshuo-Li
@wangruohui @Z-Fran @LeoXing1996 @plyfager @VongolaWu @lvhan028 @ryanxingql @gaoyang07 @ruoningYu @ChangjianZhao @zxczrx123 @jackghosts @liuwenran @CCODING04 @vansin @RoseZhao929 @shaocongliu @LigZhong @liangzelong @yanniangu
[Docs] update contribution guidance (#1527)
e8eb1be 
* [Refactor] Refactor components, losses

* [Improve] Update name

* [Fix] Update matting metrics

* [Refactor] VFI

* [Refactor] finish & rearrange all matting pipeline

* [Refactor] PSNR, SNR, MAE, MSE

* [Refactor] SSIM

* [Refactor] Inpainting pipeline

* rename

* [Refactor] train/tests tools

* [Refactor] Linear LR Schduler

* [Refactor] NIQE

* Support strict=False

* [Refactor] ReduceLR

* [Fix] add register_all_modules in utils

* [Refactor] add default runtime to all tasks

* [Refactor] Update transforms for matting

* [Refactor] Add SetValues

* [Fix] Fix test for old flavr_net

* [Improve] lasy init file_client to enable uri infer

* [Improve] Update train/test tools

* [Refactor] Matting models, components, metrics, transforms

* [Improve] Move auto-2d-to-3d to loading

* Update dataset to support null info in ann file

* Update Metrics

* Update reduce_lr

* [Refactor] BaseModel

* [Refactor] Mattor Configs

* [Refactor] Update format and data_sample

* [Refactor] VFI models and configs

* [Fix] fileclient bug in loading

* [Refactor] generate assistant transforms

* [Refactor] SRCNN, EDSR, RDN

* [Refactor] Dataset configs for inpainting

* [Fix] VFI: update param_scheduler, hooks, and epoch_base_runner

* [Fix] Hotfix segfault in CI

rm gitlab_ci

* [Refactor] EditOptimWrapperConstructor

* [Refactor] 2.0 visualization hook and sample codes

* [Fix] Add docstrings for visualizers

* [Improve] Better GCA models & ceph support for matting transforms

* [Refactor] LIIF DIC TTSR

del vis_data

* [Refactor] Merge inpainting components

* [Improve] split REDS

* [Refactor] Global&Local inpaintor

* [Refactor] Video SR (#960)

* Clear UT

* Update tof

* Update

* Update other methods

* Update all

* Update

* Update

* Fix typo

* Delete tmp file

* [Refactor] Update readme of configs (#963)

* [Docs] Add GPU info to README (#972)

Co-authored-by: 张劲东 <PJLAB\[email protected]>

* [Refactor] Refactor all inpaintors (#973)

* [Refactor] LIIF DIC TTSR

* [Refactor] Merge inpainting components

* [Improve] split REDS

* [Refactor] Global&Local inpaintor

* [Refactor] Video SR (#960)

* Clear UT

* Update tof

* Update

* Update other methods

* Update all

* Update

* Update

* Fix typo

* debug:-

* ftr: deepfillv1-local-test

* debug: deepfillv1-train

* -

* ftr: runnable train.py

* -

* -

* rebasing

* debug:-

* ftr: deepfillv1-local-test

* debug: deepfillv1-train

* -

* ftr: runnable train.py

* -

* -

* restore restoreres

* fix

* remove images

* ftr: deepfillv2 pass local test

* debug:pconv-stage1

* ftr: pconv pass local test

* dev: aot

* bugfix: aot-load-mask

* debug:pconv-train

* debug:pconv-train

* debug:pconv-train

* debug:pconv-train

* debug:aot-train

* debug:aot-train

* debug:aot-train

* refactor: refactor test script for inpainting models

* bugfix: constructing optimizers manually instead of build_optimizers[deprecated]

* bugfix: docstring coverage in pconv_inpaintor

Co-authored-by: Yshuo-Li <[email protected]>
Co-authored-by: wangruohui <[email protected]>
Co-authored-by: ys-li <[email protected]>
Co-authored-by: wangruohui <[email protected]>

* Update Metafiles for ``GPU Info`` (#974)

Update metafile

* [Update] Remove `mmcls` in requirements (#979)

* [Fix] fix model registry and docs in demo (#975)

* [Fix] fix model registry and docs in demo

* [Fix] fix matting demo and restoration image demo

* [Fix] fix inpainting demo

* [Fix] modify init_model method in inference demo

* [Fix] fix inference api

* [Fix] fix image restoration demo

* [Fix] fix video restoration and interpolation demo

* [Fix] fix video interpolation demo

* [Docs] Update Chinese Demo docs

* [Improve] Update MMCV_MAX to 2.0 (#978)

* [Improve] Update MMCV and MMENGINE version

* [Refactor] Update configs and dosctrings (#968)

1. Update docstring
2. Fix configs

* [Fix] Fix bugs found in benchmark regression  (#981)

* [Feature] scripts for benchmark regression

* [Fix] optim_scheduler in edsr_x4c64b16_g1_300k_div2k.py

* [Fix] remove ceph path from configs

* -

* clean

* restore bg_dir ceph

Co-authored-by: wangruohui <[email protected]>

* Fix gca (#988)

* fix

* fix

* remove debug

* [Fix] clear personal ceph settings in configs

* Update for publish and benchmark

* [Fix] remove personal ceph information from save_dir in configs

Co-authored-by: zenggyh1900 <[email protected]>

* [Feature] Use Circle-CI instead of Github-CI (#994)

* [Feature] support circle ci
* [Fix] Fix UT of metrics (#991)
* [Fix] Fix UT of metrics
* [Fix] add forward_tensor function in srgan (#992)

Co-authored-by: LeoXing1996 <[email protected]>
Co-authored-by: ys-li <[email protected]>

* [Fix] Using sequence_length instead of max_frame_num in video dataset and update related UT (#997)

* [Fix] Using sequence_length instead of max_frame_num in video dataset
* [Fix] Update related UT for video dataset 

Co-authored-by: LeoXing1996 <[email protected]>

* [Fix] Fix forward_train() in realbasicvsr

* [Fix] Fix bugs caused by refactoring to pass CI (#1000)

* [Fix] using tensor.repeat instead of tensor.tile due to updated torch version

* [Refactor] remove github ci

* [Fix] fix register_all_modules (#1022)

* [Refactor] refactor dir and fix bugs (#1037)

* [Refactor] clear tests_old

* [Feature] add .pre-commit0config-zh-cn.yaml

* [Refactor] expand algorithms under configs

* [Refactor] refactor config dir

* [Refactor] refactor tools dir

* [Refactor] remove components, synthesizers in mmedit/models

* [Refactor] evaluation/evaluator

* [Refactor] refactor models

* [Refactor] expand all models under models/

* [Refactor] dev scripts

* [Refactor] organize uts according to models

* [Refactor] refact utils and evaluation

* [Refactor] clear ceph paths

* [Feature] update cosinerestartlr in configs

* [Fix] Resolve some import errors (#1007)

resolve import error

Co-authored-by: LeoXing1996 <[email protected]>

* [Fix] Revise ut for image super resolution models (#1008)

[Refactor] revise UT for ISR, Matting, and VFI,

* [Refactor] Refactor VSR models (#1010)

* Refactor models and add UTs for VSR models

* [refactor] clear datasets

* refactor: remove logger

* [Refactor] add UT for common components

* [Refactor] clear unused dataset

* [refactor] update ut

* [Refactor]: clear builder
* [Feature] enable resume/amp option in train.py

* [Refactor] add UT

* fix ut

* [Refactor] expand mask and matting folder in transforms

* [Refactor] clear common, components

* update ut

* fix ut

* [Fix] Refactor evluation, engine, visualization (#1012)

* refactor

* Leoxing/refactor more (#1013)

* resolve conflict

* rename transforms/utils.py to transforms/trans_utils.py

* [fix] fix the batch size to mini-batch size in inpainting's configs

* [fix] clear personal ceph paths in configs

* [Refactor] refactor directory and fix UT (#1014)

* [Refactor] refactor configs path

* [fix] update error message in mmedit/__init__.py

* clear empty folder

* [Fix] fix data config and some import errors (#1015)

* [Fix] Fix param_scheduler of edsr x4 (#1016)

* [Fix] Fix glean (#1017)

* [Fix] Fix tof configs (#1019)

* [Fix] fix configs error

* [Feature] enable skip, rerun, and specific GPU numbers in train_benchmark 
* [Fix] fix config errors

* [Enhancement] Revise test benchmark and download script (#1021)

* revise test benchmark and download script

* hot fix

* support job watcher

* remove task name in download

* [Feature] Update dev_scripts and fix configs

* fix train_benchmark

* [Fix] dev_scripts and configs (#1026)

* fix dev_scripts and configs

* [Fix] Update config of glean (#1027)

* [Fix] Fix config of real-esrgan (#1031)

* [Fix] Fix config of RDN (#1029)

* [Fix] update config of Tdan (#1032)

* Fix config of real-esrgan

* Update config of TDAN

* [Fix] add UT for each module accordingly (#1033)

* fix test_datasets UT

* [Fix] fix test_engine UT

* [Fix] fix test_structures UT

* fix test_evaluation UT

* fix test_data_preprocessors and test_base_models UT

* fix test_models_utils UT

* fix test_editors UT

* fix test_layers UT

* fix import errors

* Add test data

* fix test_api UT

* [Fix] fix bugs in model and configs due to refactor (#1035)

* fix configs using div2k

* fix config

* fix loading pconv_mask

* fix ema hook in real_esrgan

* [Fix] fix deepfillv1 and add ut for ema

Co-authored-by: LeoXing1996 <[email protected]>
Co-authored-by: LeoXing1996 <[email protected]>
Co-authored-by: Z-Fran <[email protected]>
Co-authored-by: ys-li <[email protected]>

* [Fix] fix bug in crop transform (#1042)

* [Refactor] rename configs and update readme accordingly (#1041)

* update config_filename, content inside config files, link in readme 
* refactor apis used mmcv.cnn before

* [Fix] fix configs and OptimWrapper called in train_step (#1045)

* [Fix] config and liif (#1046)

* [Fix] Fix test benchmark (#1040)

* [Feature ] support testing with multiple datasets in test stage
* [Feature] enable testing benchmark regression by scripts

* [Fix] fix fully functional demo and tools (#1047)

* [Fix] fix fully functional demo and tools

* [Fix] config filename (#1048)

* rename config filename

* update config link in readme

* reset experiment name and base in config

* fix UT

* [Refactor] refactor documentations (#1039)

* refact dirs

* update director structure

* refactor directory structure

* [Docs] fix docs (#1049)

* [Docs] fix advanced_guides

* [Docs] Add notes docs

* [Docs] fix user_guides and model_zoo

* [Docs] migration

* [Doc] fix reviewed docs

* fix reviewd docs

* [Doc] refactor advanced_guides/model (#1061)

* [Doc] refactor advanced_guides/model

* [Doc] refactor advanced_guides/model

* doc: models

* Apply suggestions from code review

Co-authored-by: Yifei Yang <[email protected]>

* + loss.md

* fix lint

Co-authored-by: Yifei Yang <[email protected]>
Co-authored-by: Z-Fran <[email protected]>

* [Documentation] useful tools: flops, model converters (#1059)

useful tools: flops, model converters

* [Document] user_guides/train_test.md  (#1058)

* train_test: train and test with cpu/single-gpu/multi-gpu/slurm

* fix outdated message

* [Documentation] user_guides: inference (#1060)

* user_guides: inference

* fix typo

* get_started: installation and demo (#1057)

* get_started: installation and demo

* use only a demo in get_started

* update demo in get_started

Co-authored-by: Z-Fran <[email protected]>

* [Refactor] overview (#1055)

* overview

* refactor overview

* fix typo

* Apply suggestions from code review

Co-authored-by: Yifei Yang <[email protected]>

Co-authored-by: Yifei Yang <[email protected]>
Co-authored-by: Z-Fran <[email protected]>

* [Doc] fix doc building (#1066)

* advoid conflict

* fix lint

Co-authored-by: Z-Fran <[email protected]>
Co-authored-by: Yifei Yang <[email protected]>

* [Refactor] refactor data flow and engine library (#1054)

* [Refactor]: refactor data_flow and the import from mmengine
* [Fix] update UT

* [Improve] Use relpath for NIQE params (#1002)

* niqe_relpath

* [Fix] import os

Co-authored-by: zenggyh1900 <[email protected]>

* [Fix] fix training configuration in the configs of basicvsr_pp (#1067)

* -

* fix rebase conflict

* [Docs] fix docstring (#1062)

* [Doc] fix editors docsring (#1069)

Co-authored-by: Yanhong Zeng <[email protected]>

* [Doc] Refactor Chinese docs (#1070)

* [Doc] Refactor Chinese docs

* fix switch_language doc

* [Doc] fix README

* [Doc] fix README (#1071)

* [Doc] fix README

* fix readme_zh-CN

* [Fix] fix docs/en (#1072)

* remove unnecessary images and update gitignore

* autoformat docstring by docformatter

* fix link

* update switch_language

* update changelog

* Bump to version V1.0.0rc0 (#1073)

* [Refactor] update requirements and version

* remove mmengine version requirement

* [Fix] Deploy 1.x to pypi  (#1075)

[Fix] push to pypi

* [Fix] fix data flow in multi_loop (#1081)

* [Fix] fix multi_loops UT and clean debug code (#1083)

* [Fix] fix multi_loops UT and clean debug code

* [Fix] Fix concat visualizer due to data_flow refactor (#1082)

* [Fix] expand output dict for concat visualizer

* [Feature] Update Github workflow and CircleCi (#1086)

* update GitHub workflow and ci

* [Fix] fix visualization (#1091)

* [Fix] fix visualization

* fix test_visualization_hook

* [Fix] fix glean test_pipeline (#1088)

* [Doc] Add task and quickstart info in README (#1106)

* [Doc] Add task and quickstart info in README

* [Feature] Support models of MMGeneration to MMEditing (#1105)

* [Document] merge mmgeneration into mmediting  (#1111)

* merge overview, get_started, user_guides, advanced_guides
* update index.rst
* update migration and notes

* [Fix] Add Unit tests (#1109)

* update ut
Co-authored-by: zengyh1900 <[email protected]>

* [Doc] user_guides/5_visualization (#1112)

* [Documentation] update intallation scripts (#1113)

* Update 2_get_started.md (#1116)

* [Doc] fix docstring warnings (#1120)

* [Doc] fix migration docs (#1115)

* [Doc] Modify advanced guideds and notes (#1118)

* add loss table

* enhance transform and dataset docs

* change projects format

* [Doc] user_guides/2_dataset_prepare (#1122)

* [Doc] user_guides/2_dataset_prepare

* [fix] update dataset zoo

Co-authored-by: zenggyh1900 <[email protected]>

* [Doc] advanced_guides/5_metrics (#1117)

* [Doc] advanced_guides/5_metrics

* move metrics from advanced_guides to user_guides

Co-authored-by: zenggyh1900 <[email protected]>

* [Fix] Fix Windows CI (#1123)

* rename / delete some uts

* skip test ppl

* remove linux ut to speed up debug

* skip some stylegan1 uts in win-cu CI

* skip & mock some unit tests

* add debug in workflow

* fix ut for vis_backend

* skip two stage

* skip two stage encoder decoder

* skip stylegan ut

* skip plain ut

* skip more uts

* uncomment pr-stage-test.yml

* Update pr_stage_test.yml

Co-authored-by: Yanhong Zeng <[email protected]>

* [Doc] advanced-guides/1_models (#1130)

* [Doc] model_zoo (#1128)

* [Doc] model_zoo

* update number of checkpoints

Co-authored-by: zenggyh1900 <[email protected]>

* [Fix] fix warning (#1129)

fix warning

* [Enhancement] Add unit test for GAN models (#1125)

* add unit test for SAGAN

* add unit test for DCGAN

* [Fix] Skip unit test of Eq when torch < 1.8.0 (#1131)

skip unit test of Eq when torch < 1.8.0

* [Enhancement] Add dataset unit test (#1114)

* add unit test for basic conditional dataset

* add unit test for categories

* add unit test for imagenet dataset

* add unit test for cifar10 dataset

* [Enhancement] Add unit test for evaluator (#1119)

* add unit test for evaluator

* modify mock usage

* [Enhancement] Add unit test for loops + IterTimeHook (mmedit/engine/*.py) (#1121)

* add unit test for loops

* add unit test for iter time hook

* [Enhancement] Add unit tests for BaseGAN and BaseConditionalGAN (#1124)

* complete ut for baseGAN

* complete unit tests for conditional GAN

* Review by Bohuai (#1094)

* Review by Bohuai
In the example, config path is wrong.
* rebase on 1.x
Co-authored-by: zenggyh1900 <[email protected]>

* Review by Wu Bohuai (#1076)

update note

Co-authored-by: zenggyh1900 <[email protected]>

* Fix warning (#1132)

* fix: docstring warning

* [Documentation ]update model zoo (#1133)

* update model zoo

* fix lint

* [Enhancement] Add more unit tests (#1134)

* add unit test of FID inception

* revise ut of FID-inception

* add unit test for sampler, misc and visHook

* add unit test for styleGAN2 and ADA

* add unit tests for p2p

* add unit test and model for arcface

* skip windows CI for ArcFace

* skip test on pt1.6 for Ada

* fix bug

* update more unit tests

* update requirement

* fix label sample fn under windows env

* skip ut for clip loss when torch<=1.6.0

* add git to Dockerfile

* fix gpu unit test

* Remove omit when coverage is not uploaded

* skip some unit tests due to limited RAM

* [Enhancement] Add positional encoding in GANs to MMEditing (#1139)

* support mspie and pe-singan

* add configs for positional encoding in gans

* fix lint error caused by rebase

* move PESinGAN and positional encoding to MSPIE

* [Doc] update task label in README (#1137)

* [Doc] update losses.md (#1141)

* fix typo (#1140)

* [Enhancement] Revise file structure and add ut (#1145)

* refactor mmedit/models/utils/

* rename layers to base_archs

* move fid-inception and inception_utils to evaluation/funcitonal

* merge batch process to edit data processor

* move misc to utils

* rename loops.py to gen_loops.py

* move log_processor to runner

* move trans utils to utils

* move some function from metric_utils to mmedit/utils

* add unit test for inception utils.py

* add unit test for average model

* add unit test for base translation model

* add more ut for sagan generator

* add more unit test for biggan generator

* add unit test for io_utils and revise the name of variable

* add more unit test for biggan deep generator

* remove useless ut files and revise some uts

* add unit test for stylegan3

* add more unit test for cyclegan

* add more unit test for stylegan2 module

* skip unit test of CLIP loss on win-cuda env

* omit ops under stylegan2 folder

* fix ut of alpha.py to avoid the randomness

* update base_archs

* update base_archs

* update ut

* fix import error

* add unit test for singan-modules

* add unit test for sn module

* add unit test for stylegan1 and stylegan3

* revise unit test of arcFace

* add unit test for wgan-module

* complete unit test of gen metric

* remove useless unit test files

* remove useless uts and revise ut checking script

* revise omit file list in CI configs

* update dev_scripts

Co-authored-by: zenggyh1900 <[email protected]>

* [Docs] Fix Readme according to the merge (#1142)

* modify readme for merged edit

* fix readme of models

* fix lint

* fix as comments

* [Doc] Update configs in README (#1138)

* [Doc] Update configs in README

* [Fix] fix checkpoint of metafile

* [Doc] fix checkpoint in README

* [Doc] fix glean metafile

* [Doc] fix checkpoint url

* [Fix] fix model_zoo script

* [Doc] update model_zoo

Co-authored-by: Yanhong Zeng <[email protected]>

* [Doc] update dataset_zoo (#1143)

* [Doc] update dataset README
* [Doc] update daaset_zoo scripts
* [Doc] update dataset_zoo

* [Fix] Fix stylegan3 ut (#1146)

* skip some s3 ut

* fix lint

* [Fix] Fix lint after merge (#1148)

* fix lint

* avoid randomness in stylegan2 generator unit test

* [Fix] remove ToTensor transform (#1144)

* [Fix] remove ToTensor transform
* fix api inference bug
Co-authored-by: Yanhong Zeng <[email protected]>

* [Fix] replace lr_config by param_scheduler (#1147)

* [Fix] replace lr_config by param_scheduler

* fix edvr

* fix basicvsr

* remove fixed lr_config

* [Fix] fix trans_utils import error (#1149)

* [Fix] readme (#1152)

* update readme

* update readme

* [Enhancement] Add CPU unit test for IconVSR (#1151)

add cpu for iconvsr

* update model zoo

* update changelog (#1155)

* [Enhancement] Update setup and publish workflow (#1153)

* update setup and publish workflow

* install pytorch in workflow

* [Feature] Add pickle data hook (#1156)

* add pickle data hook

* fix ut

* [Fix] Fix merge stage test (#1157)

fix merge stage test

* [Fix] Fix PyTorch installation in merge stage test (#1158)

fix merge stage test

* [Improve] add a version option in docs menu (#1162)

* update docs menu

* update docs/datasets

* [Enhance] update dev_scripts for link checking (#1164)

update dev_scripts for link checking

* [Refactoring] decompose the implementations of different metrics into several files (#1161)

* refactor metrics
* add UT for refactored metrics

* [Fix] Fix PPL bug (#1172)

fix ppl bug

* [Fix] Fix some known bugs. (#1200)

* fix bug in ceph config generation

* fix download error in io_utils

* trigger CI

* [Fix] Benchmark related bugs (#1236)

* fix benchmark url and related stuff

* fix lint

* [Enhancement] Support rerun failed or canceled jobs in `train_benchmark.py` (#1259)

* trigger CI

* fix cache dir in job watcher

* support rerun failure or canceled jobs in train benchmark

* add use case and readme for rerun

* avoid trigger circle CI pr-stage-test  when .dev_script is modified

* support change cpus-per-task in train benchmark

* [Fix]  Fix bugs in `sr test config`, `realbasicvsr config` and `pconv config` (#1167)

* fix pconv bug

* fix realbasicvsr

* fix sisr test config

* [Fix] fix test of Vid4 datasets bug (#1293)

* [Fix] fix test of Vid4 datasets bug

* Update tof_x4_official_vimeo90k.py

* Update tdan_x4_1xb16-lr1e-4-400k_vimeo90k-bd.py

* Update tdan_test_config.py

* [Feature] Support multi-metrics with different sample-model (#1171)

support multi-metrics with different sample-model

* [Fix] fix GenerateSegmentIndices ut (#1302)

* [Enhancement] Reduce the randomness in unit test of `stylegan3_utils.py` (#1306)

make s3-utils unit test more robustic

* [CI] Fix GitHub windows CI (#1320)

* add 'python -m' to windows' workflow

* [Fix] fix basicvsr++ mirror sequence bug (#1304)

* [Fix] fix sisr-test psnr config (#1319)

* [Fix] fix vsr models pytorch2onnx (#1300)

* [Bug] Ensure the output type of `GenerateFacialHeatmap` is `np.float32` (#1310)

* trigger CI

* ensure output of GenerateFacialHeatmap is float32

* [Bug] Fix sampling behavior of `unpaired_dataset.py` and  urls in cyclegan's README  (#1308)

* trigger CI

* fix download url in cyclegan readme

* fix bug in unpaired dataset

* support evaluate multi metrics in one configs for cyclegan models

* [README] Fix TTSR's README (#1325)

fix TTSR's README

* [CI] Update `paths-ignore` for GitHub CI (#1327)

update paths-ignore for GitHub CI

* [Bug] Save gt images in PGGAN's `forward` (#1328)

* save gt image to data samples in forward

* add ut to ensure gt img is saved

* [Bug] Correct RDN number of channels (#1332)

* [Bug] Revise flip transformation in some conditional gan's setting (#1331)

revise flip transformation in some conditional gan's setting

* [Unit Test] Fix unit test of SNR (#1335)

fix unit test of SNR

* [Bug] Revise flavr config (#1336)

revise flavr config

* [Fix] fix realesrgan ema (#1341)

* [Fix] fix realesrgan ema

* fix

* fix ut

* fix config

* fix ut

* [Fix] Fix bugs find during benchmark running (#1348)

* fix stylegan1 register name, add fid metric mapping

* fix lint

* [Fix] fix liif test config (#1353)

* [Enhancement] Complete save_best in configs (#1349)

complete save_best in generative models and glean

* [Config] Revise discriminator's learning rate of TTSR to align with 0.x version (#1352)

align ttsr-gan's disc lr with 0.x version (1e-4 -> 1e-5)

* [Fix] fix edsr configs (#1367)

fix edsr

* [Enhancement] Add pixel value clip in visualizer (#1365)

add pixel value clip in visualizer

* [Bug] Fix randomness in FixedCrop + add L1 loss in Pix2Pix (#1364)

* fix randomness in FixedCrop + add L1 loss in Pix2Pix

* add deterministic training for p2p

* [Fix] fix realbasicvsr config (#1358)

* [Enhancement] Fix PESinGAN-inter-pad setting + add SinGAN Dataset + add SinGAN demo (#1363)

* add singan dataset

* adopt singan config with PackEditInputs

* revise forward logic of SinGANMSGeneratorPE

* fix ema-related logic of SinGAN

* add singan demo

* add unit test for pe-singan

* add unit test for singan-ema and revise singan

* [Fix] fix types of exceptions in demos (#1372)

fix types of exceptions in demos

* [Enhancement] Support deterministic training in benchmark (#1356)

* support deterministic training in benchmark

* add kill-on-bad-exit to benchmark

* [Fix] Avoid cast int and float in GenDataPreprocessor (#1385)

avoid cast int and float in GenDataPreprocessor

* [Config] Update metric config in ggan (#1386)

* update metric config in ggan

* update gen_default_runtime

* [Config] Revise batch size in wang-gp's config (#1384)

revise batch size in wang-gp's config

* [Fix]: add type and change default number of preprocess_div2k_dataset.py (#1380)

* [Fix]: add type and change default number of preprocess_div2k_dataset.py

* using mmengine.scandir instead of mmcv.scandir

Co-authored-by: zenggyh1900 <[email protected]>

* [Feature] Support qualitative comparison tools (#1303)

* Add base gui
* support patch compare
* support slider compare
* readme
* add recording prompt
* fix lint
* fix comments

Co-authored-by: unknown <[email protected]>

* [Docs] Revise docs (change PackGenInputs and GenDataSample to mmediting ones) (#1382)

revise docs (change PackGenInputs and GenDataSample to mmediting ones)

Co-authored-by: Yanhong Zeng <[email protected]>

* [Config] Revise Pix2Pix edges2shoes config (#1391)

* revise config for p2p-edge2shoes

* update batch size setting

* rename config name and revise readme

* [Bug] fix rdn and srcnn train configs (#1392)

* fix srcnn

* fix rdn

* [Fix] Fix test/val pipeline of pegan configs (#1393)

fix ppegan configs

* [Fix] Modify Readme of S3 (#1398)

modify readme

* [Fix] Correct fid of ggan (#1397)

fix readme of ggan

* [Feature] support instance_aware_colorization inference (#1370)

* add Instance-aware Image Colorization
* update configs
* update docstring if instance-aware image colorization
* add unit test of instance_aware_colorization
* use mmdet
Co-authored-by: ruoning <[email protected]>

* bump to v1.0.0rc2 (#1405)

* [Improve] add a version option in docs menu (#1162)
* [Enhance] update dev_scripts for link checking (#1164)
* [Refactoring] decompose the implementations of different metrics into several files (#1161)
* [Fix] Fix PPL bug (#1172)
* [Fix] Fix some known bugs. (#1200)
* [Fix] Benchmark related bugs (#1236)
* [Enhancement] Support rerun failed or canceled jobs in `train_benchmark.py` (#1259)
* [Fix]  Fix bugs in `sr test config`, `realbasicvsr config` and `pconv config` (#1167)
* [Fix] fix test of Vid4 datasets bug (#1293)
* [Feature] Support multi-metrics with different sample-model (#1171)
* [Fix] fix GenerateSegmentIndices ut (#1302)
* [Enhancement] Reduce the randomness in unit test of `stylegan3_utils.py` (#1306)
* [CI] Fix GitHub windows CI (#1320)
* [Fix] fix basicvsr++ mirror sequence bug (#1304)
* [Fix] fix sisr-test psnr config (#1319)
* [Fix] fix vsr models pytorch2onnx (#1300)
* [Bug] Ensure the output type of `GenerateFacialHeatmap` is `np.float32` (#1310)
* [Bug] Fix sampling behavior of `unpaired_dataset.py` and  urls in cyclegan's README  (#1308)
* [README] Fix TTSR's README (#1325)
* [CI] Update `paths-ignore` for GitHub CI (#1327)
* [Bug] Save gt images in PGGAN's `forward` (#1328)
* [Bug] Correct RDN number of channels (#1332)
* [Bug] Revise flip transformation in some conditional gan's setting (#1331)
* [Unit Test] Fix unit test of SNR (#1335)
* [Bug] Revise flavr config (#1336)
* [Fix] fix realesrgan ema (#1341)
* [Fix] Fix bugs find during benchmark running (#1348)
* [Fix] fix liif test config (#1353)
* [Enhancement] Complete save_best in configs (#1349)
* [Config] Revise discriminator's learning rate of TTSR to align with 0.x version (#1352)
* [Fix] fix edsr configs (#1367)
* [Enhancement] Add pixel value clip in visualizer (#1365)
* [Bug] Fix randomness in FixedCrop + add L1 loss in Pix2Pix (#1364)
* [Fix] fix realbasicvsr config (#1358)
* [Enhancement] Fix PESinGAN-inter-pad setting + add SinGAN Dataset + add SinGAN demo (#1363)
* [Fix] fix types of exceptions in demos (#1372)
* [Enhancement] Support deterministic training in benchmark (#1356)
* [Fix] Avoid cast int and float in GenDataPreprocessor (#1385)
* [Config] Update metric config in ggan (#1386)
* [Config] Revise batch size in wang-gp's config (#1384)
* [Fix]: add type and change default number of preprocess_div2k_dataset.py (#1380)
* [Feature] Support qualitative comparison tools (#1303)
* [Docs] Revise docs (change PackGenInputs and GenDataSample to mmediting ones) (#1382)
* [Config] Revise Pix2Pix edges2shoes config (#1391)
* [Bug] fix rdn and srcnn train configs (#1392)
* [Fix] Fix test/val pipeline of pegan configs (#1393)
* [Fix] Modify Readme of S3 (#1398)
* [Fix] Correct fid of ggan (#1397)
* [Feature] support instance_aware_colorization inference (#1370)

Co-authored-by: ruoning <[email protected]>
Co-authored-by: Yifei Yang <[email protected]>
Co-authored-by: LeoXing1996 <[email protected]>
Co-authored-by: Z-Fran <[email protected]>
Co-authored-by: Qunliang Xing <[email protected]>
Co-authored-by: Yang Gao <[email protected]>
Co-authored-by: ruoning <[email protected]>

* [Bug] fix cain config (#1404)

* fix srcnn

* [Bug] fix cain config

* [Fix] Revise config and pretrain model loading in esrgan (#1407)

revise config and pretrain model loading in esrgan

* [Fix] Fix lsgan config (#1409)

fix lsgan config

* [Enhancement] Support `try_import` for `mmdet` (#1408)

* add try-import for mmdet

* revise import logic

* add unit test for try_import

Co-authored-by: Yanhong Zeng <[email protected]>
Co-authored-by: Yifei Yang <[email protected]>

* [Enhancement] Set ``real_feat`` to cpu in inception_utils (#1415)

fix s2 configs

* [Enhancement] git ignore slurm generated files (#1416)

* improve git ignore

* add a blank line

* [Fix] modify readme and configs of stylegan2&pegan (#1418)

modify readme and configs of stylegan2&pegan

* [Enhancement] Support try-import for `clip` (#1420)

* support try-import for clip

* update ci

* [Enhancement]: Improve the rendering of Docs-API (#1373)

* [Enhancement]: Improve the rendering of Docs-API
* fix lint

Co-authored-by: root <[email protected]>
Co-authored-by: Yanhong Zeng <[email protected]>

* [Fix] Complete requirements (#1419)

* add requirements

* fix lint

Co-authored-by: Yanhong Zeng <[email protected]>

* [Doc] Update changelog and README for 1.0.0rc3. (#1421)

* update changelog and README

* revise change log

* [Bug] Install clip in merge stage test (#1423)

install clip in merge stage test

* [Fix] Install clip in windows CI (#1424)

install clip in windows CI

* Bump to 1.0.0rc3. (#1425)

* [Improve] add a version option in docs menu (#1162)

* update docs menu

* update docs/datasets

* [Enhance] update dev_scripts for link checking (#1164)

update dev_scripts for link checking

* [Refactoring] decompose the implementations of different metrics into several files (#1161)

* refactor metrics
* add UT for refactored metrics

* [Fix] Fix PPL bug (#1172)

fix ppl bug

* [Fix] Fix some known bugs. (#1200)

* fix bug in ceph config generation

* fix download error in io_utils

* trigger CI

* [Fix] Benchmark related bugs (#1236)

* fix benchmark url and related stuff

* fix lint

* [Enhancement] Support rerun failed or canceled jobs in `train_benchmark.py` (#1259)

* trigger CI

* fix cache dir in job watcher

* support rerun failure or canceled jobs in train benchmark

* add use case and readme for rerun

* avoid trigger circle CI pr-stage-test  when .dev_script is modified

* support change cpus-per-task in train benchmark

* [Fix]  Fix bugs in `sr test config`, `realbasicvsr config` and `pconv config` (#1167)

* fix pconv bug

* fix realbasicvsr

* fix sisr test config

* [Fix] fix test of Vid4 datasets bug (#1293)

* [Fix] fix test of Vid4 datasets bug

* Update tof_x4_official_vimeo90k.py

* Update tdan_x4_1xb16-lr1e-4-400k_vimeo90k-bd.py

* Update tdan_test_config.py

* [Feature] Support multi-metrics with different sample-model (#1171)

support multi-metrics with different sample-model

* [Fix] fix GenerateSegmentIndices ut (#1302)

* [Enhancement] Reduce the randomness in unit test of `stylegan3_utils.py` (#1306)

make s3-utils unit test more robustic

* [CI] Fix GitHub windows CI (#1320)

* add 'python -m' to windows' workflow

* [Fix] fix basicvsr++ mirror sequence bug (#1304)

* [Fix] fix sisr-test psnr config (#1319)

* [Fix] fix vsr models pytorch2onnx (#1300)

* [Bug] Ensure the output type of `GenerateFacialHeatmap` is `np.float32` (#1310)

* trigger CI

* ensure output of GenerateFacialHeatmap is float32

* [Bug] Fix sampling behavior of `unpaired_dataset.py` and  urls in cyclegan's README  (#1308)

* trigger CI

* fix download url in cyclegan readme

* fix bug in unpaired dataset

* support evaluate multi metrics in one configs for cyclegan models

* [README] Fix TTSR's README (#1325)

fix TTSR's README

* [CI] Update `paths-ignore` for GitHub CI (#1327)

update paths-ignore for GitHub CI

* [Bug] Save gt images in PGGAN's `forward` (#1328)

* save gt image to data samples in forward

* add ut to ensure gt img is saved

* [Bug] Correct RDN number of channels (#1332)

* [Bug] Revise flip transformation in some conditional gan's setting (#1331)

revise flip transformation in some conditional gan's setting

* [Unit Test] Fix unit test of SNR (#1335)

fix unit test of SNR

* [Bug] Revise flavr config (#1336)

revise flavr config

* [Fix] fix realesrgan ema (#1341)

* [Fix] fix realesrgan ema

* fix

* fix ut

* fix config

* fix ut

* [Fix] Fix bugs find during benchmark running (#1348)

* fix stylegan1 register name, add fid metric mapping

* fix lint

* [Fix] fix liif test config (#1353)

* [Enhancement] Complete save_best in configs (#1349)

complete save_best in generative models and glean

* [Config] Revise discriminator's learning rate of TTSR to align with 0.x version (#1352)

align ttsr-gan's disc lr with 0.x version (1e-4 -> 1e-5)

* [Fix] fix edsr configs (#1367)

fix edsr

* [Enhancement] Add pixel value clip in visualizer (#1365)

add pixel value clip in visualizer

* [Bug] Fix randomness in FixedCrop + add L1 loss in Pix2Pix (#1364)

* fix randomness in FixedCrop + add L1 loss in Pix2Pix

* add deterministic training for p2p

* [Fix] fix realbasicvsr config (#1358)

* [Enhancement] Fix PESinGAN-inter-pad setting + add SinGAN Dataset + add SinGAN demo (#1363)

* add singan dataset

* adopt singan config with PackEditInputs

* revise forward logic of SinGANMSGeneratorPE

* fix ema-related logic of SinGAN

* add singan demo

* add unit test for pe-singan

* add unit test for singan-ema and revise singan

* [Fix] fix types of exceptions in demos (#1372)

fix types of exceptions in demos

* [Enhancement] Support deterministic training in benchmark (#1356)

* support deterministic training in benchmark

* add kill-on-bad-exit to benchmark

* [Fix] Avoid cast int and float in GenDataPreprocessor (#1385)

avoid cast int and float in GenDataPreprocessor

* [Config] Update metric config in ggan (#1386)

* update metric config in ggan

* update gen_default_runtime

* [Config] Revise batch size in wang-gp's config (#1384)

revise batch size in wang-gp's config

* [Fix]: add type and change default number of preprocess_div2k_dataset.py (#1380)

* [Fix]: add type and change default number of preprocess_div2k_dataset.py

* using mmengine.scandir instead of mmcv.scandir

Co-authored-by: zenggyh1900 <[email protected]>

* [Feature] Support qualitative comparison tools (#1303)

* Add base gui
* support patch compare
* support slider compare
* readme
* add recording prompt
* fix lint
* fix comments

Co-authored-by: unknown <[email protected]>

* [Docs] Revise docs (change PackGenInputs and GenDataSample to mmediting ones) (#1382)

revise docs (change PackGenInputs and GenDataSample to mmediting ones)

Co-authored-by: Yanhong Zeng <[email protected]>

* [Config] Revise Pix2Pix edges2shoes config (#1391)

* revise config for p2p-edge2shoes

* update batch size setting

* rename config name and revise readme

* [Bug] fix rdn and srcnn train configs (#1392)

* fix srcnn

* fix rdn

* [Fix] Fix test/val pipeline of pegan configs (#1393)

fix ppegan configs

* [Fix] Modify Readme of S3 (#1398)

modify readme

* [Fix] Correct fid of ggan (#1397)

fix readme of ggan

* [Feature] support instance_aware_colorization inference (#1370)

* add Instance-aware Image Colorization
* update configs
* update docstring if instance-aware image colorization
* add unit test of instance_aware_colorization
* use mmdet
Co-authored-by: ruoning <[email protected]>

* [Bug] fix cain config (#1404)

* fix srcnn

* [Bug] fix cain config

* [Fix] Revise config and pretrain model loading in esrgan (#1407)

revise config and pretrain model loading in esrgan

* [Fix] Fix lsgan config (#1409)

fix lsgan config

* [Enhancement] Support `try_import` for `mmdet` (#1408)

* add try-import for mmdet

* revise import logic

* add unit test for try_import

Co-authored-by: Yanhong Zeng <[email protected]>
Co-authored-by: Yifei Yang <[email protected]>

* [Enhancement] Set ``real_feat`` to cpu in inception_utils (#1415)

fix s2 configs

* [Enhancement] git ignore slurm generated files (#1416)

* improve git ignore

* add a blank line

* [Fix] modify readme and configs of stylegan2&pegan (#1418)

modify readme and configs of stylegan2&pegan

* [Enhancement] Support try-import for `clip` (#1420)

* support try-import for clip

* update ci

* [Enhancement]: Improve the rendering of Docs-API (#1373)

* [Enhancement]: Improve the rendering of Docs-API
* fix lint

Co-authored-by: root <[email protected]>
Co-authored-by: Yanhong Zeng <[email protected]>

* [Fix] Complete requirements (#1419)

* add requirements

* fix lint

Co-authored-by: Yanhong Zeng <[email protected]>

* [Doc] Update changelog and README for 1.0.0rc3. (#1421)

* update changelog and README

* revise change log

* [Bug] Install clip in merge stage test (#1423)

install clip in merge stage test

* [Fix] Install clip in windows CI (#1424)

install clip in windows CI

Co-authored-by: Yanhong Zeng <[email protected]>
Co-authored-by: Yifei Yang <[email protected]>
Co-authored-by: Z-Fran <[email protected]>
Co-authored-by: Qunliang Xing <[email protected]>
Co-authored-by: Yang Gao <[email protected]>
Co-authored-by: ruoning <[email protected]>
Co-authored-by: unknown <[email protected]>
Co-authored-by: root <[email protected]>

* [Doc] fix readthedoc requirements (#1437)

* [Fix] fix srgan train config (#1441)

* fix srcnn

* [Fix] fix srgan train config

* [Enhancement] Fix 'after_test_ite' of vis hook + support 'max_save_at_test' (#1430)

* fix 'after_test_ite' of vis hook + support 'max_save_at_test'

* delete comment in unit test

* [Feature] Support NAFNet model  (#1369)

* NAFNet feature

* NAFNet and Real-ESRGAN

* NAFNet for denoising cfg

* Modify NAFNet Gopro Config

* NAFNet Deblurring Config

* Debug removal

* NAFNet Feature

* NAFNet Feature Support

* NAFNet Feature Support

* NAFNet code check finished

* NAFNet unit tests

* NAFNet unit test

* Docstring for NAFNet

* NAFNet unit tests

* NAFNet

* NAFNet unit tests

* NAFNet unit tests

* PSNRLoss

* NAFNet unit tests

* NAFNet Readme

* add NAFNet into __init__

* NAFNet baseline

* NAFNet cfgs for test

* NAFNet readme

* NAFNet Readme and Configs

* Rename files

* BaseModule

* clear undesired comments

* NAFNet Readme

* NAFNet Readme

* NAFNet components

* NAFNet UTs

Co-authored-by: Yanhong Zeng <[email protected]>

* [Doc] fix readthedoc requirements (#1448)

* [Doc] fix readthedoc requirements

* fix readthedoc

* [Fix] Revise unit test of log processor (#1459)

revise unit test of log processor

* [Refactor] Replace UnconditionalImageDataset with BasicImageDataset (#1456)

replace uncond to basic

Co-authored-by: LeoXing1996 <[email protected]>

* [Docs] Refine dev-1.x docs (#1461)

* fix bug

* fix bug-v2

* Update docs/en/2_get_started.md

Co-authored-by: 赵昌健 <[email protected]>
Co-authored-by: Yifei Yang <[email protected]>

* [Doc] Refine User guides (#1462)

* update user_guides doc

* fix lint

Co-authored-by: ruixiangzhang <[email protected]>
Co-authored-by: plyfager <[email protected]>

* update user guides (#1463)

Co-authored-by: Yifei Yang <[email protected]>

* [Enhancement] Refactor high-level APIs (#1410)

* add high-level inference api and run conditional model good.

* [high-level api] run unconditional model.

* [high-level api] add matting inferencer.

* [high-level api] add inpainting inferencer.

* [high-level api] add translation inferencer.

* [high-level api] add restoration inferencer.

* [high-level api] add inference functions

* [high-level api] add video interpolation inference

* [high-level api] add video restoration inferencer

* [high-level api] pass linter check

* [hight-level api] append  linter check

* [high-level api] delete old interface py code

* [high-level api] remove unused code.

* [high-level api] add comments for inferences.

* [high-level api] delete unused parameters and add extra parameters.

* [high-level api] add inference tutorial.

* [high-level api] add unit test for inference_functions

* [high-level api] fix path error.

* [high-level api] delete old unit test file.

* [high-level api] add ut for edit and conditional inferncer.

* [high-level api] fix edit ut bug and add ut of base_inference, inpainting.

* [high-level api] fix edit.py ut and add two uts.

* [high-level api] add result_out_dir to UTs and add two new uts.

* [high-level api] fix unconditional ut out of ram and add two uts.

* [high-level api] try to satisfy ut code coverage.

* [high-level api] use pytest.raises to catch ut error.

* [high-level api] add more test case in inference_functions

* [high-level api] add video restoration test case.

* [high-level api] video interpolation support dir input output and add ut case

* [high-level api] video restoration support input_dir and add uts.

* [high-level api] add more uts for inference funcs.

* [high-level api] fix bug in inference_functions

* [high-level api] add more uts.

* [high-level api] make colorization inference be tested.

* [high-level api] roll back last commit.

* [high-level api] delete unused code.

* [high-level api] remove default value.

* [high-level api] rename version to setting.

* [high-level api] add examples in edit.py and remove duplicated funcs.

* [high-level api] add log for functions  not used.

* [high-level api] load mean std from cfg and use basedataelement

* [high-level api] do unittest with cu102 version.

* [high-level api] add more uts.

* [high-level api] revert change in da2557b

* [high-level api] add uts.

* [high-level api] replace ckpt to http url./

* [high-level api] read config from metafile and download ckpt automatically.

* [high-level api] make default setting all to 0.

* [high-level api] add content in ipynb

* [high-level api] add ut for test_edit modification.

* [high-level api] fix task in scripts and metafiles

* [high-level api] read task name from metafile

* [high-level api] reproduce nafnet metafile

* [high-level api] put good example to inference ipynb

* [high-level api] revert inpainting demo to global local.

* [high-level api] add model setting and extra parameters in ipynb

* [high-level api] add README.md and more instructions in ipynb.

* [high-level api] append to last commit.

* [high-level api] fix typo.

* [high-level api] add ut to refresh building.

* [high-level api] fix readme review comments.

* [high-level api] refresh readme.

* [high-level api] refresh demo readme and fix typo

* [high-level api] resolve review comment

* [high-level api] fix type.

* [high-level api] revert change.

* [high-level api] misc change.

* [high-level api] fix type again.

Co-authored-by: liuwenran <[email protected]>

* [Fix] fix tdan train config (#1457)

* fix srcnn

* [Fix] fix tdan train config

* add 'DIM online merge config demo' (#783)

* add 'DIM online merge config demo'

* rename

* fix config

Co-authored-by: Z-Fran <[email protected]>
Co-authored-by: Z-Fran <[email protected]>

* [Fix] Adopt ut with newest mmengine (#1471)

adopt ut with newest mmengine

* [Doc] update qq group link (#1473)

* [bug] Update base_conditional_gan.py (#1474)

Update base_conditional_gan.py

* [bug] Update inference_functions.py (#1470)

* Update inference_functions.py

remove repeated const

* pass linter check.

Co-authored-by: rangoliu <[email protected]>

* [bug] fix 1.x face demo readme config error (#1469)

Update 3_inference.md

config

* [Fix] fix pconv train config (#1467)

* fix srcnn

* [Fix] fix pconv train config

* rename

* [Fix] fix tdan gpu info (#1486)

* fix srcnn

* [Fix] fix tdan gpu info

* merge dev-1.x

* [Fix] Wrong path resulsting in pre-commit error in Python 3.9 (#1488)

fix relative path error in `update_model_index.py`

* [Fix] fix basicvsr train config (#1485)

* fix srcnn

* [Fix] fix basicvsr train config

* [Feature] Support disco-diffusion text-2-image (#1234)

* going through adm unconditional sampling

* add config and code for test

* fix lint

* modify unet

* format adm

* support adm 512

* fix lint

* support cls-g sampling

* support ddim sampling

* init disco

* support disco-diffusion text2image

* support secondary model in disco-diffusion (#1368)

* support init_image as input

* init docstring

* refactor disco

* fix lint

* fix lint

* remove disco bug

* remove data_preprocessor

* complete docstring of disco and partial guider

* complete docstring for guider

* refine secondary model

* fix lint

* move cutter and loss config to infer

* fix adm and unet

* rename config

* support portrait generator config

* fix clip wrapper

* move unet to DDPM

* rename clip_ext

* adjust requirements

* try_import

* add dist.get_rank() == 0 as additional condition

* add resize_right to requirements

* remove disco_baseline

* update url

* fix a disco typo

* add  imagenet 256 config

* Make Disco's readme simple

* rename disco to disco_diffusion

* add adm readme

* fix lint

* support directly init disco with instance module

* add ut of disco

* fix init

* fix lint

* improve docstring coverage

* fix lint

* fix docstring

* fix lint

* add credits

* mv losses

* fix lint

* rename diffuser

* fix lint

* delete null

* rm raise error

* fix comment

Co-authored-by: yanniangu <[email protected]>

* [Feature] Support EG3D (#1494)

* [Enhancement] Support conditional input for StyleGAN modules (#1443)

* support conditional input for stylegan v2 discriminator + Mapping Network

* revise docstring

* add unit test for stylegan2 discriminator + revise some variable name

* add unit test for stylegan3's mapping network

* support no noise injection in ModConv + support no upsample in ModToRGB

* revise some variable name in stylegan3's mapping network

* support conditional input in stylegan2's generator and add unit test

* revise unit test of stylegan2's generator

* cover more line in stylegan3's mapping network

* cover more line in stylegan2's module

* remove some comment from stylegan2's generator

* revise add noise behvior of stylegan2's generator

* revise typing hint of base cond gan

* revise as comment

* rename 'cond' to 'label' to align with conditional GAN's interface + revise some docstring

* support update w_avg with ema in StyleGAN2's generator

* fix bug in w_avg's EMA

* support update_ws arg in StyleGAN2's forward

* revise stylegan2 and stylegan3 as comment

* revise stylegan3 module

* revise behavior of pixel norm in StyleGAN2's generator

* revise stylegan2 and stylegan3 as comment

* [Feature] Support conditional FID for EG3D (#1447)

* support conditional FID for EG3D

* remove useless code form conditional FID

* add unit test for CondFid

* fix fid

* import condFID in mmedit.evaluation.__init__

* remove conditional FID and support need_cond in GenerativeMetrics

* remove force_recal_inception

* revise fid

* rename need_cond to need_cond_input as comment

* add log when need_cond_input is True + not convert cond type to torch.float32

* [Feature] Support camera class for EG3D (#1446)

* add camera for eg3d

* revise docstring of camera

* add unit test for camera

* revise uniform sampling in camera

* revise camera as comment

* move normalize_vecs to tensor_utils

* [Feature] Support EG3D renderer (#1454)

* support EG3D renderer

* revise renderer + add unit test for renderer

* revise render and math_utils as comment

* revise docstring of EG3DRenderer

* [Feature] Support ray sampler for EG3D (#1455)

* support ray sampler

* add unit test for ray sampler

* adopt meshgrid for different torch version

* [Feature] Add EG3D modules (#1458)

* add EG3D modules

* disable antialias in F.interpolation to adopt torch<1.11

* remove useless comment from EG3D module

* [Feature] Support TriPlane Generator for EG3D. (#1464)

* add EG3D's generator

* support render_kwargs in TriPlaneGenerator's forward

* support randomize_noise in eg3d generator

* revise comment for eg3d generator

* reduce the number of sampling points to reduce test time

* support change hidden channels and output channels for super resolution model

* [Feature] Add eg3d discriminator (#1493)

* add eg3d discriminator

* rename eg3d_discriminator to dual_discriminator

* revert init file for eg3d-disc

* adopt discriminator's interpolation to differert torch version

* [Feature] Support EG3D model and High-level API (#1482)

* support EG3D model

* support high-level API for EG3D

* add docstring for eg3d.interpolation

* adopt EG3D with revised camera

* revise demo + support user defined arguments for high-level API

* support set random seed in high-level API

* add EG3D to __init__

* update readme

* rename math-utils to eg3d-utils + complete docstring for EG3D model

* resort init

* adopt interpolation with lower version of torch

* add imageio-ffmpeg to requirements

* add configs for EG3D

* update f-string in edit

* move imageio-ffmpeg to optional.txt and add try-import for imageio

* install imageio-ffmpeg in CI

* add model checkpoint url to README

* revise demo README + rename num-frames to num-images

* revise configs

* [Fix] Support arccos in camera for torch <= 1.6 (#1498)

* support arccos in camera for torch <= 1.6

* skip some ut on windows CI

* revise unit tests

* [Enhanchment] Fix unit test of EG3D Render for torch < 1.8.0 (#1499)

adopt torch.nan_to_num for torch < 1.8.0 + mock torch version in unit test to cover more lines

* [Enhancement] Mock LPIPS module in PPL metric (#1490)

* mock LPIPS module in PPL metric

* mock cpu test

* [feature] add t2i infer. (#1504)

[t2i infer] add t2i infer.

Co-authored-by: liuwenran <[email protected]>

* [Docs] Add a notebook for Disco Diffusion (#1497)

* support disco notebook

* fix disco weights url

* fix tutorials notebook

* add tutorials to readme

* Fix tutorials for colab

* fix colab of disco

* fix installing mmedit

* fix notebook

* fix clip wrapper

* fix readme

* rename config

* fix lint

* prepare for release

* [Enhancement] Add EG3D and Disco to README (#1505)

* prepare for release

* fix comments

* fix readme

* fix readme

* fix readme zh

* Fix disco notebook (#1507)

fix notebook

* [Enhancement] Fix unit test for disco diffusion (#1511)

fix unit test for disco diffusion

* [Enhancement] Add torch1.13 checking in CI (#1509)

add torch1.13 checking in CI

* [Enhance] set the default 404 page for readthedocs (#1513)

* enable 404 extension

* update conf.py

* fix extensions

* add notfoundpage in conf

* update requirements

* fix typo

* [Enhanchement] Revise test requirements and CI (#1514)

* revise test requirements and CI

* remove open-clip from tests.txt

* [Enhance] Recursive generate summary and docstring for API Reference (#1517)

* remove apis

* enable recursive api rendering

* fix template

* fix rendering for models

* fix lint

* using autoapi instead of autodoc

* use autoapi instead of autodocs

* fix lint

Co-authored-by: zengyh1900 <[email protected]>

* [Fix] fix documentation link checker (#1522)

fix incorrect command

* [Doc] improve the structure of documentation (#1521)

refactor doc structure

* [Doc] improve toctree and update model zoo and dataset zoo (#1525)

* update circleci

* update model zoo

* update model zoo

* move changelog.md

* fix lint

* fix index toctree

* [Enhance] enable projects  (#1526)

* update projects

* fix typo

* fix typo

* update workflow: lint_only for projects

* update community

Co-authored-by: Yshuo-Li <[email protected]>
Co-authored-by: wangruohui <[email protected]>
Co-authored-by: wangruohui <[email protected]>
Co-authored-by: ys-li <[email protected]>
Co-authored-by: Z-Fran <[email protected]>
Co-authored-by: 张劲东 <PJLAB\[email protected]>
Co-authored-by: LeoXing1996 <[email protected]>
Co-authored-by: LeoXing1996 <[email protected]>
Co-authored-by: Yifei Yang <[email protected]>
Co-authored-by: Z-Fran <[email protected]>
Co-authored-by: VongolaWu <[email protected]>
Co-authored-by: lvhan028 <[email protected]>
Co-authored-by: Qunliang Xing <[email protected]>
Co-authored-by: Yang Gao <[email protected]>
Co-authored-by: ruoning <[email protected]>
Co-authored-by: unknown <[email protected]>
Co-authored-by: ruoning <[email protected]>
Co-authored-by: root <[email protected]>
Co-authored-by: ChangjianZhao <[email protected]>
Co-authored-by: 赵昌健 <[email protected]>
Co-authored-by: Ruixiang Zhang <[email protected]>
Co-authored-by: ruixiangzhang <[email protected]>
Co-authored-by: Jack <[email protected]>
Co-authored-by: rangoliu <[email protected]>
Co-authored-by: liuwenran <[email protected]>
Co-authored-by: CCODING <[email protected]>
Co-authored-by: Z-Fran <[email protected]>
Co-authored-by: vansin <[email protected]>
Co-authored-by: RoseZhao929 <[email protected]>
Co-authored-by: shaocongliu <[email protected]>
Co-authored-by: TheBlackBox <[email protected]>
Co-authored-by: liangzelong <[email protected]>
Co-authored-by: yanniangu <[email protected]>
Co-authored-by: zengyh1900 <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@zengyh1900 zengyh1900 zengyh1900 approved these changes

Assignees

@Z-Fran Z-Fran

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@Z-Fran @zengyh1900