无法将Python文件MNIST导入USPS Google Pixel DA

2024-05-14 04:25:08 发布

您现在位置:Python中文网/ 问答频道 /正文

目前正在尝试运行谷歌的像素级域适配模型。我们能够运行MNIST到MNIST-M的训练和测试。在datasets文件夹中添加了USPS tfrecords,我们将MNIST\u M.py文件重新创建到美国药典.py文件。导入文件就像导入MNIST一样。不知何故,当我们运行培训时,架构无法找到USPS文件。你知道吗

谷歌最初的模式:Google's Model

正在运行的命令:

bazel run domain_daptation/pixel_domain_adaptation:pixeldatrain -- --dataset_dir $DSN_DATA_DIR --source_dataset mnist --target_dataset usps

位置美国药典.py文件:

/home/../models/domain\u自适应/数据集:usps 原始堆栈跟踪:

INFO: Analysed target //domain_adaptation/pixel_domain_adaptation:pixelda_train (2 packages loaded).
INFO: Found 1 target...
Target //domain_adaptation/pixel_domain_adaptation:pixelda_train up-to-date:
bazel-bin/domain_adaptation/pixel_domain_adaptation/pixelda_train
INFO: Elapsed time: 0.575s, Critical Path: 0.12s
INFO: Build completed successfully, 1 total action

INFO: Running command line: bazel-bin/domain_adaptation/pixel_domain_adaptation/pixelda_train --dataset_dir /home/gabe/models/domain_adaptation/datasets --source_dataset mnist --target_dataset usps
Traceback (most recent call last):
File "/home/.../.cache/bazel/_bazel_root/b15fa6ac5bc49187f8190f3d0425c8db/execroot/__main__/bazel-out/k8-fastbuild/bin/domain_adaptation/pixel_domain_adaptation/pixelda_train.runfiles/__main__/domain_adaptation/pixel_domain_adaptation/pixelda_train.py", line 24, in <module>
from domain_adaptation.datasets import dataset_factory
File "/home/.../.cache/bazel/_bazel_root/b15fa6ac5bc49187f8190f3d0425c8db/execroot/__main__/bazel-out/k8-fastbuild/bin/domain_adaptation/pixel_domain_adaptation/pixelda_train.runfiles/__main__/domain_adaptation/datasets/dataset_factory.py", line 26, in <module>
from domain_adaptation.datasets import usps
ImportError: cannot import name usps
ERROR: Non-zero return code '1' from command: Process exited with status 1

遵循其中一个用户的建议,将位置添加到域\自适应/数据集/构建。我又执行了同样的命令。你知道吗

py_library(
    name = "dataset_factory",
    srcs = ["dataset_factory.py"],
    deps = [
        ":usps",
        "//slim:mnist",
        ],
)

新堆栈跟踪:

ERROR: /home/.../models/domain_adaptation/datasets/BUILD:23:12: in deps              attribute of py_library rule //domain_adaptation/datasets:dataset_factory: rule '//domain_adaptation/datasets:usps' does not exist
ERROR: Analysis of target '//domain_adaptation/pixel_domain_adaptation:pixelda_train' failed; build aborted: Analysis of target '//domain_adaptation/datasets:dataset_factory' failed; build aborted
INFO: Elapsed time: 0.219s
FAILED: Build did NOT complete successfully (0 packages loaded)
ERROR: Build failed. Not running target

Tags: 文件pyinfotargethomedomainfactorytrain