Skip to content
This repository has been archived by the owner on Dec 1, 2021. It is now read-only.

Commit

Permalink
Move lmnet/lmnet/environment.py to blueoil/environment.py (#832)
Browse files Browse the repository at this point in the history
  • Loading branch information
iizukak authored Feb 13, 2020
1 parent c4018fc commit bbe6c54
Show file tree
Hide file tree
Showing 31 changed files with 35 additions and 35 deletions.
2 changes: 1 addition & 1 deletion blueoil/cmd/convert_weight_from_darknet.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
import numpy as np
import tensorflow as tf

from lmnet import environment
from blueoil import environment
from blueoil.networks.classification.darknet import Darknet
from blueoil.networks.object_detection.yolo_v2 import YoloV2
from blueoil.utils import config as config_util
Expand Down
2 changes: 1 addition & 1 deletion blueoil/cmd/evaluate.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
import click
import tensorflow as tf

from lmnet import environment
from blueoil import environment
from blueoil.datasets.base import ObjectDetectionBase
from blueoil.datasets.dataset_iterator import DatasetIterator
from blueoil.datasets.tfds import TFDSClassification, TFDSObjectDetection
Expand Down
2 changes: 1 addition & 1 deletion blueoil/cmd/export.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
import numpy as np
import tensorflow as tf

from lmnet import environment
from blueoil import environment
from blueoil.utils.image import load_image
from blueoil.utils import config as config_util
from blueoil.utils import executor
Expand Down
2 changes: 1 addition & 1 deletion blueoil/cmd/measure_latency.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
import tensorflow as tf
from tensorflow.python.client import device_lib

from lmnet import environment
from blueoil import environment
from blueoil.utils.image import load_image
from blueoil.utils import config as config_util
from blueoil.utils import executor
Expand Down
2 changes: 1 addition & 1 deletion blueoil/cmd/output_event.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
from tensorboard.backend.event_processing.event_accumulator import EventAccumulator
from tensorboard.backend.event_processing.io_wrapper import GetLogdirSubdirectories

from lmnet import environment
from blueoil import environment


def _get_metrics_keys(event_accumulator):
Expand Down
2 changes: 1 addition & 1 deletion blueoil/cmd/profile_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
import click
import tensorflow as tf

from lmnet import environment
from blueoil import environment
from blueoil.utils import config as config_util
from blueoil.utils import executor

Expand Down
2 changes: 1 addition & 1 deletion blueoil/datasets/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@

import numpy as np

from lmnet import environment
from blueoil import environment


class Base(metaclass=ABCMeta):
Expand Down
File renamed without changes.
2 changes: 1 addition & 1 deletion blueoil/utils/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
from tensorflow.io import gfile
from yaml.representer import Representer

from lmnet import environment
from blueoil import environment
from blueoil.common import Tasks
from lmnet.data_processor import Processor, Sequence

Expand Down
2 changes: 1 addition & 1 deletion blueoil/utils/executor.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
import tensorflow as tf
from tensorflow.io import gfile

from lmnet import environment
from blueoil import environment


def init_logging(config):
Expand Down
2 changes: 1 addition & 1 deletion lmnet/executor/predict.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
import numpy as np
import tensorflow as tf

from lmnet import environment
from blueoil import environment
from blueoil.utils.image import load_image
from blueoil.utils import config as config_util
from blueoil.utils.executor import search_restore_filename
Expand Down
2 changes: 1 addition & 1 deletion lmnet/executor/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@
from tensorflow.core.util.event_pb2 import SessionLog
from tensorflow.keras.utils import Progbar

from lmnet import environment
from blueoil import environment
from blueoil.common import Tasks
from blueoil.datasets.base import ObjectDetectionBase
from blueoil.datasets.dataset_iterator import DatasetIterator
Expand Down
2 changes: 1 addition & 1 deletion lmnet/tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
import pytest
import tensorflow as tf

from lmnet import environment
from blueoil import environment


@pytest.fixture
Expand Down
2 changes: 1 addition & 1 deletion lmnet/tests/executor_tests/test_build_tfds.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@

from blueoil.cmd.build_tfds import run
from executor.train import run as train_run
from lmnet import environment
from blueoil import environment
from blueoil.datasets.dataset_iterator import DatasetIterator
from blueoil.datasets.tfds import TFDSClassification, TFDSObjectDetection
from blueoil.utils import config as config_util
Expand Down
2 changes: 1 addition & 1 deletion lmnet/tests/executor_tests/test_export.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@

from blueoil.cmd.export import run
from executor.train import run as train_run
from lmnet.environment import setup_test_environment
from blueoil.environment import setup_test_environment

# Apply reset_default_graph() in conftest.py to all tests in this file.
# Set test environment
Expand Down
2 changes: 1 addition & 1 deletion lmnet/tests/executor_tests/test_predict.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@

from executor.predict import run
from executor.train import run as train_run
from lmnet.environment import setup_test_environment
from blueoil.environment import setup_test_environment

# Apply reset_default_graph() in conftest.py to all tests in this file.
# Set test environment
Expand Down
2 changes: 1 addition & 1 deletion lmnet/tests/executor_tests/test_profile.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@

from blueoil.cmd.profile_model import run
from executor.train import run as train_run
from lmnet.environment import setup_test_environment
from blueoil.environment import setup_test_environment

# Apply reset_default_graph() in conftest.py to all tests in this file.
# Set test environment
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@ def test_dataset_iterator_batch_order():


if __name__ == '__main__':
from lmnet import environment
from blueoil import environment
environment.setup_test_environment()
test_dataset_iterator_batch_size()
test_dataset_iterator_batch_order()
2 changes: 1 addition & 1 deletion lmnet/tests/lmnet_tests/datasets_tests/test_delta_mark.py
Original file line number Diff line number Diff line change
Expand Up @@ -253,7 +253,7 @@ def test_delta_mark_object_detection_has_validation_path():


if __name__ == '__main__':
from lmnet.environment import setup_test_environment
from blueoil.environment import setup_test_environment
setup_test_environment()

test_delta_mark_classification()
Expand Down
4 changes: 2 additions & 2 deletions lmnet/tests/lmnet_tests/datasets_tests/test_image_folder.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
import numpy as np
import pytest

from lmnet import environment
from blueoil import environment
from blueoil.datasets.dataset_iterator import DatasetIterator
from blueoil.datasets.image_folder import ImageFolderBase
from lmnet.pre_processor import Resize
Expand Down Expand Up @@ -125,7 +125,7 @@ def test_image_folder_onthefly():


if __name__ == '__main__':
from lmnet.environment import setup_test_environment
from blueoil.environment import setup_test_environment
setup_test_environment()

test_image_folder()
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -198,7 +198,7 @@ def test_custom_has_validation_open_images_v4_object_detection():


if __name__ == '__main__':
from lmnet.environment import setup_test_environment
from blueoil.environment import setup_test_environment
setup_test_environment()

test_open_images_v4_classification()
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from easydict import EasyDict

from executor.train import start_training
from lmnet import environment
from blueoil import environment
from blueoil.common import Tasks
from blueoil.networks.classification.darknet import Darknet
from blueoil.datasets.image_folder import ImageFolderBase
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from easydict import EasyDict

from executor.train import start_training
from lmnet import environment
from blueoil import environment
from blueoil.common import Tasks
from blueoil.networks.classification.lm_resnet import LmResnetQuantize
from blueoil.datasets.image_folder import ImageFolderBase
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from easydict import EasyDict

from executor.train import start_training
from lmnet import environment
from blueoil import environment
from blueoil.common import Tasks
from blueoil.networks.classification.lmnet_quantize import LmnetQuantize
from blueoil.datasets.image_folder import ImageFolderBase
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
import pytest
import tensorflow as tf

from lmnet import environment
from blueoil import environment
from blueoil.common import Tasks
from blueoil.networks.keypoint_detection.lm_single_pose_v1 import LmSinglePoseV1Quantize
from blueoil.datasets.mscoco_2017 import MscocoSinglePersonKeypoints
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
from easydict import EasyDict

from executor.train import start_training
from lmnet import environment
from blueoil import environment
from blueoil.common import Tasks
from blueoil.networks.object_detection.yolo_v1 import YoloV1
from blueoil.datasets.pascalvoc_2007 import Pascalvoc2007
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
from easydict import EasyDict

from executor.train import start_training
from lmnet import environment
from blueoil import environment
from blueoil.common import Tasks
from lmnet.data_processor import Sequence
from blueoil.networks.object_detection.yolo_v2 import YoloV2
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from easydict import EasyDict

from executor.train import start_training
from lmnet import environment
from blueoil import environment
from blueoil.common import Tasks
from blueoil.networks.object_detection.yolo_v2_quantize import YoloV2Quantize
from blueoil.datasets.pascalvoc_2007 import Pascalvoc2007
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
from easydict import EasyDict

from executor.train import start_training
from lmnet import environment
from blueoil import environment
from blueoil.common import Tasks
from lmnet.data_processor import Sequence
from blueoil.networks.segmentation.lm_bisenet import LMBiSeNet
Expand Down
2 changes: 1 addition & 1 deletion lmnet/tests/test_configs.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@

import pytest

from lmnet import environment
from blueoil import environment
from blueoil.utils.config import _load_py, check_config, save_yaml

pytestmark = pytest.mark.usefixtures("set_test_environment")
Expand Down
10 changes: 5 additions & 5 deletions tests/e2e/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
from blueoil.cmd.convert import convert
from blueoil.cmd.predict import predict
from blueoil.cmd.train import train
from lmnet import environment
from blueoil import environment
from blueoil.utils.config import load


Expand All @@ -23,7 +23,7 @@ def init_env():
environment_originals = {}
environ_originals = {}

# TODO: Remove this setting after lmnet.environment has been refactored.
# TODO: Remove this setting after blueoil.environment has been refactored.
envs = {
"DATA_DIR": os.path.join(blueoil_dir, "lmnet", "tests", "fixtures", "datasets"),
"OUTPUT_DIR": train_output_dir.name,
Expand Down Expand Up @@ -69,15 +69,15 @@ def run_all_steps(dirs, config_file):
config = load(config_path)

# Train
# TODO: Remove this setting after lmnet.environment has been refactored.
# TODO: Remove this setting after blueoil.environment has been refactored.
environment._init_flag = False
experiment_id, checkpoint_name = train(config_path)

train_output_dir = os.path.join(dirs["train_output_dir"], experiment_id)
assert os.path.exists(os.path.join(train_output_dir, 'checkpoints'))

# Convert
# TODO: Remove this setting after lmnet.environment has been refactored.
# TODO: Remove this setting after blueoil.environment has been refactored.
environment._init_flag = False
convert(experiment_id)

Expand All @@ -102,7 +102,7 @@ def run_all_steps(dirs, config_file):
predict_input_dir = os.path.join(dirs["blueoil_dir"], "lmnet/tests/fixtures/sample_images")
predict_output_dir = dirs["predict_output_dir"]

# TODO: Remove this setting after lmnet.environment has been refactored.
# TODO: Remove this setting after blueoil.environment has been refactored.
environment._init_flag = False
predict(predict_input_dir, predict_output_dir, experiment_id, checkpoint_name)

Expand Down

0 comments on commit bbe6c54

Please sign in to comment.