From 87735acb6c7796a07c3656424f8a0c5fc91487b8 Mon Sep 17 00:00:00 2001 From: Turgut Date: Tue, 22 Jun 2021 13:37:06 +0400 Subject: [PATCH] Update create_d2go.py ``` python create_d2go.py Traceback (most recent call last): File "create_d2go.py", line 16, in from d2go.tests.data_loader_helper import LocalImageGenerator, register_toy_dataset ModuleNotFoundError: No module named 'd2go.tests.data_loader_helper' ``` --- D2Go/create_d2go.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/D2Go/create_d2go.py b/D2Go/create_d2go.py index 3dc326d6..9c424442 100644 --- a/D2Go/create_d2go.py +++ b/D2Go/create_d2go.py @@ -13,7 +13,7 @@ from d2go.model_zoo import model_zoo from mobile_cv.common.misc.file_utils import make_temp_directory -from d2go.tests.data_loader_helper import LocalImageGenerator, register_toy_dataset +from d2go.utils.testing.data_loader_helper import LocalImageGenerator, _register_toy_dataset patch_d2_meta_arch() @@ -33,13 +33,13 @@ def create_fake_detection_data_loader(height, width, is_train): image_generator = LocalImageGenerator(image_dir, width=width, height=height) if is_train: - with register_toy_dataset( + with _register_toy_dataset( "default_dataset_train", image_generator, num_images=3 ): train_loader = runner.build_detection_train_loader(cfg) yield train_loader else: - with register_toy_dataset( + with _register_toy_dataset( "default_dataset_test", image_generator, num_images=3 ): test_loader = runner.build_detection_test_loader(