Squeezedet Keras - PullRequest
Новая
       11

Squeezedet Keras

0 голосов
/ 04 марта 2019

Я пытаюсь запустить следующий детектор объектов, https://github.com/omni-us/squeezedet-keras

Иерархия папок выглядит следующим образом:

squeezedet-keras 

   -> experiments
        -> kitti

   -> scripts
        -> train.py

   -> main
        -> model
            -> squeezeDet
            -> imagenet.h5

Теперь из squeezedet-keras / эксперименты / китти, я запускаю команду: 

>>python ../../scripts/train.py --init ../../main/model/imagenet.h5

Файл train.py начинается с этой строки: 

from main.model.squeezeDet import  SqueezeDet

Это выдает ошибку:

Трассировка (последний последний вызов):

Файл "../../scripts/train.py", строка 9, в

из main.model.squeezeDet import SqueezeDet

ModuleNotFoundError: Нет модуля с именем 'main'

Я запускаю его с virtualenv, и мой python по умолчанию - 3.7.2, на ноутбуке с Windows 10.

1 Ответ

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

Интерпретатору python трудно загружать модули проекта.

Загрузчик модулей распознает только папки, которые:

  • находятся внутри {venv_folder}/lib/python{version}/site-packages/ --- онивероятно были установлены с pip install ... или python setup.py install
  • Находятся в текущем рабочем каталоге
  • Указаны PYTHONPATH env var

Самый простой способЧтобы решить эту проблему, установите env var: 

cd working_dir/
source /path/to/my/venv/
export PYTHONPATH=/path/to/squeezedet-keras
python ../../scripts/train.py --init ../../main/model/imagenet.h5

Другой способ - просто перейти в папку проекта перед выполнением сценариев: 

cd /path/to/squeezedet-keras
source /path/to/my/venv/
python scripts/train.py --init main/model/imagenet.h5
...