如何在 google colab 中導入 torchreid？

Question

我正在嘗試安裝 torchreid，它是 PyTorch 中用於人員重新識別的庫。我已按照git 存儲庫中提到的步驟進行操作，但出現此錯誤。

#conda install 
!wget -c https://repo.anaconda.com/miniconda/Miniconda3-4.5.4-Linux-x86_64.sh
!chmod +x Miniconda3-4.5.4-Linux-x86_64.sh
!bash ./Miniconda3-4.5.4-Linux-x86_64.sh -b -f -p /usr/local

!conda install -q -y --prefix /usr/local python=3.6 ujson

import sys
sys.path.append('/usr/local/lib/python3.6/site-packages')

!git clone https://github.com/KaiyangZhou/deep-person-reid.git

!cd deep-person-reid/
!conda create --name torchreid python=3.7
!conda activate torchreid
!pip install -r /content/deep-person-reid/requirements.txt
!conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

import torchreid

錯誤

ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-1-4faf46a39b5d> in <module>()
----> 1 import torchreid

ModuleNotFoundError: No module named 'torchreid'

Answer 1

首先停用您創建的 conda 環境'torchreid'。 再次激活它並嘗試導入。

停用火炬手

conda 激活torchreid

注意：您正在將 torchreid 安裝在特定的 conda 環境中，該環境無法從該環境外部訪問。

Answer 2

運行以下命令，

# install torchreid (don't need to re-build it if you modify the source code)
python setup.py develop