Skip to content

Commit

Permalink
[release] add new model to model zoo (#1571)
Browse files Browse the repository at this point in the history
* [fix] fix ci error

* update test.

* changelog date

* add new model to model zoo

* Update test_inference_functions.py
  • Loading branch information
liuwenran authored Jan 4, 2023
1 parent d9369cb commit 740757f
Show file tree
Hide file tree
Showing 4 changed files with 18 additions and 10 deletions.
10 changes: 7 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,7 @@ hope MMEditing could provide better experience.

### 🌟 Preview of 1.x version

A brand new version of [**MMEditing v1.0.0rc5**](https://github.com/open-mmlab/mmediting/releases/tag/v1.0.0rc5) was released in 03/01/2023:
A brand new version of [**MMEditing v1.0.0rc5**](https://github.com/open-mmlab/mmediting/releases/tag/v1.0.0rc5) was released in 04/01/2023:

- Support well-known text-to-image method [Stable Diffusion](configs/stable_diffusion/README.md)!
- Support an efficient image restoration algorithm [Restormer](configs/restormer/README.md)!
Expand Down Expand Up @@ -272,23 +272,27 @@ Supported algorithms:
<details open>
<summary>Text2Image</summary>

-[GLIDE](projects/glide/configs/README.md) (NeurIPS'2021)
-[Disco-Diffusion](configs/disco_diffusion/README.md)
-[Stable-Diffusion](configs/stable_diffusion/README.md)

</details>

<details open>

<summary>3D-aware Generation</summary>

-[EG3D](configs/eg3d/README.md)
-[EG3D](configs/eg3d/README.md) (CVPR'2022)

</details>

<details open>

<summary>Image Restoration</summary>

-[NAFNet](configs/nafnet/README.md)
-[SwinIR](configs/swinir/README.md) (ICCVW'2021)
-[NAFNet](configs/nafnet/README.md) (ECCV'2022)
-[Restormer](configs/restormer/README.md) (CVPR'2022)

</details>

Expand Down
10 changes: 7 additions & 3 deletions README_zh-CN.md
Original file line number Diff line number Diff line change
Expand Up @@ -105,7 +105,7 @@ MMEditing 缜密地设计新的框架并将其精心实现,希望能够为您

### 🌟 1.x 预览版本

全新的 [**MMEditing v1.0.0rc5**](https://github.com/open-mmlab/mmediting/releases/tag/v1.0.0rc4) 已经在 03/01/2022 发布:
全新的 [**MMEditing v1.0.0rc5**](https://github.com/open-mmlab/mmediting/releases/tag/v1.0.0rc5) 已经在 04/01/2023 发布:

- 支持了著名的文本生成图像方法 [stable diffusion](configs/stable_diffusion/README.md)!
- 支持了一个高效的图像复原算法 [Restormer](configs/restormer/README.md)!
Expand Down Expand Up @@ -270,23 +270,27 @@ pip3 install -e .
<details open>
<summary>Text2Image</summary>

-[GLIDE](projects/glide/configs/README.md) (NeurIPS'2021)
-[Disco-Diffusion](configs/disco_diffusion/README.md)
-[Stable-Diffusion](configs/stable_diffusion/README.md)

</details>

<details open>

<summary>3D-aware Generation</summary>

-[EG3D](configs/eg3d/README.md)
-[EG3D](configs/eg3d/README.md) (CVPR'2022)

</details>

<details open>

<summary>Image Restoration</summary>

-[NAFNet](configs/nafnet/README.md)
-[SwinIR](configs/swinir/README.md) (ICCVW'2021)
-[NAFNet](configs/nafnet/README.md) (ECCV'2022)
-[Restormer](configs/restormer/README.md) (CVPR'2022)

</details>

Expand Down
4 changes: 2 additions & 2 deletions docs/en/changelog.md
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
# Changelog

## v1.0.0rc5 (03/01/2023)
## v1.0.0rc5 (04/01/2023)

**Highlights**
We are excited to announce the release of MMEditing 1.0.0rc4. This release supports 49+ models, 180+ configs and 177+ checkpoints in MMGeneration and MMEditing. We highlight the following new features
We are excited to announce the release of MMEditing 1.0.0rc5. This release supports 49+ models, 180+ configs and 177+ checkpoints in MMGeneration and MMEditing. We highlight the following new features

- Support Restormer
- Support GLIDE
Expand Down
4 changes: 2 additions & 2 deletions tests/test_apis/test_inferencers/test_inference_functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -216,8 +216,6 @@ def test_restoration_face_inference():
@pytest.mark.skipif(
'win' in platform.system().lower() and 'cu' in torch.__version__,
reason='skip on windows-cuda due to limited RAM.')
@pytest.mark.skipif(
sys.version_info < (3, 7), reason='skip because python version is old.')
def test_restoration_inference():
if torch.cuda.is_available():
device = torch.device('cuda', 0)
Expand All @@ -236,6 +234,8 @@ def test_restoration_inference():
assert output.detach().cpu().numpy().shape == (3, 480, 500)


@pytest.mark.skipif(
sys.version_info < (3, 7), reason='skip because python version is old.')
def test_restoration_video_inference():
if torch.cuda.is_available():
device = torch.device('cuda', 0)
Expand Down

0 comments on commit 740757f

Please sign in to comment.