diff --git a/README.md b/README.md index eaef31c9..00e98a6b 100644 --- a/README.md +++ b/README.md @@ -112,6 +112,8 @@ Major features: - Provides rich components and strategies. - Complete controls on the training process with different levels of APIs. +![mmengine_dataflow](https://github.com/open-mmlab/mmengine/assets/58739961/267db9cb-72e4-4af2-a58b-877b30091acc) + ## Installation Before installing MMEngine, please ensure that PyTorch has been successfully installed following the [official guide](https://pytorch.org/get-started/locally/). diff --git a/README_zh-CN.md b/README_zh-CN.md index cac94f9f..7f524047 100644 --- a/README_zh-CN.md +++ b/README_zh-CN.md @@ -113,6 +113,8 @@ MMEngine 是一个基于 PyTorch 实现的,用于训练深度学习模型的 - 提供了丰富的组件和策略。 - 使用不同等级的 API 控制训练过程。 +![mmengine_dataflow](https://github.com/open-mmlab/mmengine/assets/58739961/267db9cb-72e4-4af2-a58b-877b30091acc) + ## 安装 在安装 MMEngine 之前,请确保 PyTorch 已成功安装在环境中,可以参考 [PyTorch 官方安装文档](https://pytorch.org/get-started/locally/)。