VR-Baseline
Video Restoration Toolbox including FGST (ICML 2022), S2SVR (ICML 2022), etc.
File Explorer
Download Latest Version (.zip)- profiles_settings.xml
- Project_Default.xml
- misc.xml
- modules.xml
- VR-Baseline.iml
- FGST_deblur_dvd.py
- FGST_deblur_dvd_test.py
- FGST_deblur_gopro.py
- FGST_deblur_gopro_test.py
- metafile.yml
- S2SVR_deblur_gopro.py
- S2SVR_sr_reds4.py
- S2SVR_sr_vimeo.py
- DVD_test.txt
- DVD_train.txt
- GoPro_test.txt
- GoPro_train.txt
- DVD_Util.py
- GoPro_Util.py
- restoration_video_demo.py
- Dockerfile
- blur2sharp.gif
- compressed2enhanced.gif
- lr2sr.gif
- ntire.png
- __init__.py
- generation_inference.py
- inpainting_inference.py
- matting_inference.py
- restoration_face_inference.py
- restoration_inference.py
- restoration_video_inference.py
- test.py
- train.py
- video_interpolation_inference.py
- __init__.py
- eval_hooks.py
- metric_utils.py
- metrics.py
- niqe_pris_params.npz
- __init__.py
- wrappers.py
- __init__.py
- ema.py
- visualization.py
- __init__.py
- builder.py
- __init__.py
- lr_updater.py
- __init__.py
- dist_utils.py
- __init__.py
- distributed_wrapper.py
- mask.py
- misc.py
- __init__.py
- augmentation.py
- blur_kernels.py
- compose.py
- crop.py
- formating.py
- generate_assistant.py
- loading.py
- matlab_like_resize.py
- matting_aug.py
- normalization.py
- random_degradations.py
- random_down_sampling.py
- utils.py
- __init__.py
- distributed_sampler.py
- __init__.py
- base_dataset.py
- base_sr_dataset.py
- builder.py
- dataset_wrappers.py
- registry.py
- sr_folder_multiple_gt_dataset.py
- sr_reds_multiple_gt_dataset.py
- sr_vimeo90k_multiple_gt_dataset.py
- __init__.py
- basicvsr_net.py
- correlation.py
- FGST.py
- FGST_util.py
- pwclite.py
- S2SVR.py
- S2SVR_util.py
- __init__.py
- __init__.py
- aspp.py
- contextual_attention.py
- conv.py
- downsample.py
- ensemble.py
- flow_warp.py
- gated_conv_module.py
- gca_module.py
- generation_model_utils.py
- img_normalize.py
- linear_module.py
- mask_conv_module.py
- model_utils.py
- partial_conv.py
- separable_conv_module.py
- sr_backbone_utils.py
- upsample.py
- __init__.py
- gan_loss.py
- perceptual_loss.py
- pixelwise_loss.py
- utils.py
- __init__.py
- basic_restorer.py
- basicvsr.py
- __init__.py
- base.py
- builder.py
- registry.py
- __init__.py
- cli.py
- collect_env.py
- logger.py
- setup_env.py
- __init__.py
- version.py
- pwclite.tar
- docs.txt
- readthedocs.txt
- runtime.txt
- tests.txt
- test_generation_datasets.py
- test_matting_datasets.py
- test_repeat_dataset.py
- test_sr_dataset.py
- test_vfi_dataset.py
- test_augmentation.py
- test_compose.py
- test_crop.py
- test_formating.py
- test_generate_assistant.py
- test_loading.py
- test_matlab_resize.py
- test_normalization.py
- test_pipeline_utils.py
- test_random_degradations.py
- test_random_down_sampling.py
- test_trimap.py
- test_metrics.py
- test_aot_model.py
- test_decoders.py
- test_deepfill_decoder.py
- test_deepfill_encdec.py
- test_deepfill_encoder.py
- test_encoder_decoder.py
- test_encoders.py
- test_gl_model.py
- test_pconv_encdec.py
- test_generators.py
- test_basicvsr_net.py
- test_basicvsr_plusplus.py
- test_dic_net.py
- test_duf.py
- test_edvr_net.py
- test_glean_net.py
- test_iconvsr.py
- test_liif_net.py
- test_rdn.py
- test_real_basicvsr_net.py
- test_sr_backbones.py
- test_tdan_net.py
- test_tof.py
- test_cain_net.py
- test_tof_vfi_net.py
- test_common_module.py
- test_ensemble.py
- test_flow_warp.py
- test_img_normalize.py
- test_model_utils.py
- test_sampling.py
- test_deepfill_disc.py
- test_discriminators.py
- test_light_cnn.py
- test_multi_layer_disc.py
- test_unet_disc.py
- test_deepfill_refiner.py
- test_matting_refiners.py
- test_mlp_refiner.py
- test_stylegan2.py
- test_feedback_hour_glass.py
- test_lte.py
- test_aot_inpaintor.py
- test_deepfill_inpaintor.py
- test_gl_inpaintor.py
- test_one_stage_inpaintor.py
- test_pconv_inpaintor.py
- test_two_stage_inpaintor.py
- test_feature_loss.py
- test_losses.py
- test_mattors.py
- test_basic_restorer.py
- test_basicvsr_model.py
- test_dic_model.py
- test_edvr_model.py
- test_esrgan.py
- test_glean.py
- test_liif.py
- test_real_basicvsr.py
- test_real_esrgan.py
- test_srgan.py
- test_tdan.py
- test_ttsr.py
- test_cyclegan.py
- test_search_transformer.py
- test_basic_interpolator.py
- test_cain.py
- test_base_model.py
- test_apis.py
- test_dataset_builder.py
- test_ema_hook.py
- test_eval_hook.py
- test_optimizer.py
- test_visual_hook.py
- test_mask_generation.py
- test_modify_args.py
- test_onnx_wraper.py
- test_pix2pix.py
- test_setup_env.py
- test_tensor2img.py
- test_inference.py
- mmedit2torchserve.py
- mmedit_handler.py
- test_torchserver.py
- deploy_test.py
- dist_test.sh
- dist_train.sh
- evaluate_comp1k.py
- get_flops.py
- onnx2tensorrt.py
- publish_model.py
- pytorch2onnx.py
- slurm_test.sh
- slurm_train.sh
- test.py
- test_lj.py
- train.py
- CITATION.cff
- LICENSE
- MANIFEST.in
- model-index.yml
- README.md
- requirements.txt
- setup.cfg
- setup.py
# Installation Guide
1. Get the code
git clone https://github.com/linjing7/VR-Baseline
Downloads the entire project code from GitHub to your computer.
cd VR-Baseline
Moves into the project folder you just downloaded.
2. Official Install Script
Easy RecommendedPrerequisites
- Python 3 Python is required to use pip.
pip install torchvision==0.9.0 torch==1.8.0 torchaudio==0.8.0
Installs the package published on PyPI directly β no need to clone the source.
pip install openmim
Installs the package published on PyPI directly β no need to clone the source.
pip install -v -e .
Installs the package published on PyPI directly β no need to clone the source.
pip install cupy-cuda101==7.7.0
Installs the package published on PyPI directly β no need to clone the source.
After installing, open a new terminal and run the program's version command (e.g. --version) to confirm it worked.
Pulled directly from this repo's README.
3. Docker
EasyPrerequisites
- Git Needed to download the project code from GitHub.
- Docker Desktop Needed to build and run containers. Install it and keep it running in the background.
docker build -f docker/Dockerfile -t vr-baseline .
Builds a runnable image based on the Dockerfile.
docker run -p 8080:80 vr-baseline
Runs the built image as an actual container.
Run docker compose ps to check the containers are Up. If the README mentions a port, open http://localhost:PORT in your browser.
4. Python
EasyPrerequisites
pip install torchvision==0.9.0 torch==1.8.0 torchaudio==0.8.0
Installs the package published on PyPI directly β no need to clone the source.
pip install -r requirements.txt
Installs the Python libraries listed in requirements.txt (or similar).
pip install openmim
Installs the package published on PyPI directly β no need to clone the source.
pip install -v -e .
Installs the package published on PyPI directly β no need to clone the source.
pip install cupy-cuda101==7.7.0
Installs the package published on PyPI directly β no need to clone the source.
If it runs without errors and prints output in the terminal, it worked.
Pulled directly from this repo's README.
// repository documentation
Was this content helpful?
(0 ratings)