Поэтому я пытаюсь установить Deeplab в Colab.

Я бегаю:

[ 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

Из ноутбука и успешно импортирует

3
Gabe 17 Апр 2019 в 05:17

2 ответа

Лучший ответ

Решил вопрос. К сожалению, связанный репозиторий, который дал abggcv, сталкивается с той же проблемой, на которую ссылался этот вопрос.

Вы должны клонировать репо как обычно и запустить все как обычно. Единственное изменение заключается в том, что перед запуском 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/ - это путь, по которому я клонировал репо. Вам нужно будет изменить это, если каталог, в котором вы клонировали репо, отличается.

Кроме того, по какой-то причине вы не сможете запустить train.py/eval.py/vis.py подряд. Даже очистка флагов приведет к ошибке об дублирующем флаге. Чтобы это исправить, просто перезапустите среду выполнения (не потеряет ваши файлы).

Удачного сегментирования!

0
Gabe 8 Ноя 2019 в 22:04

Здесь - репозиторий Github с записной книжкой Colab, на которой работает Deeplab.
Я не проверял его, но то, как вы загрузили весь каталог на Google Диск, не подходит для запуска вещей на Colab.
Представьте, что Colab - это отдельная машина, и вы устанавливаете Google Drive на эту машину. Все, что доступно на вашем Google Диске, не обязательно доступно для машины Colab. Вам нужно будет добавить путь к папке на Google Диске (скажем, '\content\drive\My Drive\<path_to_your_folder>') в sys.path для машины Colab, используя sys.path.insert(0, <path_of_your_drive_folder>), чтобы сделать этот путь доступным для среды Python, работающей на машине Colab.

1
abggcv 12 Май 2019 в 11:40