在 colab 中设置 DeepLabV3

Posted 2023-03-27

【中文标题】在 colab 中设置 DeepLabV3

【英文标题】：Setting up DeepLabV3 in colab

【发布时间】：2019-09-07 04:40:06

【问题描述】：

所以我正在尝试在 colab 中设置 deeplab。

我正在跑步：

[1]

from google.colab import drive
drive.mount('/content/drive')
%cd /content/drive/My\ Drive/deeplab_files

[2]

%env PYTHONPATH=/content/drive/My\ Drive/deeplab_files/:/content/drive/My\ Drive/deeplab_files/slim
!echo $PYTHONPATH

[3]

!python deeplab/vis.py \
    --logtostderr \
    --vis_split="val" \
    --model_variant="xception_65" \
    --atrous_rates=6 \
    --atrous_rates=12 \
    --atrous_rates=18 \
    --output_stride=16 \
    --decoder_output_stride=4 \
    --vis_crop_size=360 \
    --vis_crop_size=480 \
    --dataset="camvid" \
    --colormap_type="pascal" \
    --checkpoint_dir='/content/drive/My\ Drive/deeplab_files/deeplab/datasets/PQR/exp/train_on_trainval_set/train' \
    --vis_logdir='/content/drive/My\ Drive/deeplab_files/deeplab/datasets/PQR/exp/train_on_trainval_set/vis' \
    --dataset_dir='/content/drive/My\ Drive/deeplab_files/deeplab/datasets/PQR/tfrecord'

然而，最后一个命令返回

sh: 1: export: Drive/deeplab_files/slim:/content/drive/My Drive/deeplab_files/:/content/drive/My Drive/deeplab_files/slim: bad variable name
Traceback (most recent call last):
  File "deeplab/vis.py", line 28, in <module>
    from deeplab import common
ModuleNotFoundError: No module named 'deeplab'

有人知道如何设置 deeplab 吗？我在我的个人机器上设置了它，但它太慢了。我将整个文件夹上传到我的 gdrive。

奇怪的是我能做到

from deeplab import common

从笔记本中成功导入

【参考方案1】：

Here 是一个 Github 存储库，其中包含一个运行 deeplab 的 Colab 笔记本。 我尚未对其进行测试，但您将整个目录上传到 Google Drive 的方式并不是在 Colab 上运行的正确方式。 将 Colab 视为一台单独的机器，并且您正在将 Google Drive 安装在这台机器上。您的 Google Drive 上可用的任何内容不一定可用于 Colab 机器。您必须使用 sys.path.insert(0, <path_of_your_drive_folder>) 将 Google Drive 文件夹的路径（例如 '\content\drive\My Drive\<path_to_your_folder>'）添加到 Colab 机器的 sys.path，以使该路径可用于在 Colab 机器上运行的 python 环境。

【参考方案2】：

解决了 mt 问题。不幸的是，abggcv 提供的链接回购遇到了这个问题所引用的相同问题。

您应该像往常一样克隆 repo，并正常运行一切。唯一的变化是，在运行 train.py、eval.py 或 vis.py 之前，您需要运行以下代码块：

%cd /root/deeplabvc/models/research/
import sys
sys.path.extend(['/root/deeplabvc/models/research/', '/root/deeplab/models/research/slim/'])

请注意，/root/deeplab/ 是我克隆存储库的路径。如果克隆 repo 的目录不同，则需要更改此设置。

此外，由于某种原因，您将无法连续运行 train.py/eval.py/vis.py。即使清除标志也会给你一个关于重复标志的错误。要解决此问题，只需重新启动运行时（不会丢失文件）。

分割愉快！

【参考方案3】：

Deeplab 导入错误主要发生在 PYTHONPATH 设置不正确时。给出的安装说明不适用于 COLAB 环境。以下对我有用

%cd /content/deeplab/models/research/
!mkdir -p deeplab/datasets/pascal_voc_seg/exp/train_on_train_set/train
!mkdir -p deeplab/datasets/pascal_voc_seg/exp/train_on_train_set/eval
!mkdir -p deeplab/datasets/pascal_voc_seg/exp/train_on_train_set/vis
!echo $PYTHONPATH
%env PATH_TO_TRAIN_DIR=/content/deeplab/models/research/deeplab/datasets/pascal_voc_seg/exp/train_on_train_set/train
%env PATH_TO_DATASET=/content/deeplab/models/research/deeplab/datasets/pascal_voc_seg/tfrecord
%env PYTHONPATH=/content/deeplab/models/research:/content/deeplab/models/research/deeplab:/content/deeplab/models/research/slim:/env/python
!echo $PYTHONPATH

Here 是我的 COLAB 笔记本，用于训练有效的 deeplab