fix installation guide

open-mmlab · May 2, 2023 · e0ef602 · e0ef602
1 parent d676daf
commit e0ef602
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -104,7 +104,7 @@ conda create -n openmmlab python=3.8 pytorch=1.10 cudatoolkit=11.3 torchvision -
 conda activate openmmlab
 pip3 install openmim
 mim install mmcv-full
-git clone https://github.com/open-mmlab/mmpose.git
+git clone https://github.com/open-mmlab/mmpose.git -b 0.x
 cd mmpose
 pip3 install -e .
 ```

diff --git a/README_CN.md b/README_CN.md
@@ -102,7 +102,7 @@ conda create -n openmmlab python=3.8 pytorch=1.10 cudatoolkit=11.3 torchvision -
 conda activate openmmlab
 pip3 install openmim
 mim install mmcv-full
-git clone https://github.com/open-mmlab/mmpose.git
+git clone https://github.com/open-mmlab/mmpose.git -b 0.x
 cd mmpose
 pip3 install -e .
 ```

diff --git a/docs/en/install.md b/docs/en/install.md
@@ -67,7 +67,7 @@ mim install mmcv-full
 Case a: If you develop and run mmpose directly, install it from source:
 
 ```shell
-git clone https://github.com/open-mmlab/mmpose.git
+git clone https://github.com/open-mmlab/mmpose.git -b 0.x
 cd mmpose
 pip install -r requirements.txt
 pip install -v -e .

diff --git a/docs/zh_cn/install.md b/docs/zh_cn/install.md
@@ -80,7 +80,7 @@ mim install mmcv-full
 这种情况下，从源码按如下方式安装 mmpose：
 
 ```shell
-git clone https://github.com/open-mmlab/mmpose.git
+git clone https://github.com/open-mmlab/mmpose.git -b 0.x
 cd mmpose
 pip install -r requirements.txt
 pip install -v -e .

diff --git a/mmpose/datasets/pipelines/shared_transform.py b/mmpose/datasets/pipelines/shared_transform.py
@@ -2,7 +2,6 @@
 import warnings
 from collections.abc import Sequence
 
-import albumentations
 import mmcv
 import numpy as np
 import torch
@@ -13,6 +12,11 @@
 
 from ..builder import PIPELINES
 
+try:
+    import albumentations
+except ImportError:
+    albumentations = None
+
 
 @PIPELINES.register_module()
 class ToTensor: