Настройка DeepLabV3 в Colab - PullRequest
Новая
       35

Настройка DeepLabV3 в Colab

0 голосов
/ 17 апреля 2019

Итак, я пытаюсь настроить 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

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

1 Ответ

0 голосов
/ 12 мая 2019

Здесь - это репозиторий 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.

...