AssertionError for tfrecord_dir in stylegan .什么是正确的论点？

Question

当我尝试运行样式training.py时，它会抛出AssertionError，如下所示。正确的tfrecord_dir参数应该是什么？

在training.py中有；

desc += '-dataset';     
dataset = EasyDict(tfrecord_dir='dataset', resolution=128); 
train.mirror_augment = False

我把tfrecord_dir论点说成是dataset。这条路应该是正确的。我的tf记录器路径是\stylegan\dataset。我的training.py路径是\stylegan。

这是我的错误：

PS C:\Users\dell> & C:/Users/dell/AppData/Local/Programs/Python/Python310/python.exe "c:/Users/dell/Desktop/Data Science/GAN/stylegan/train.py"

File "c:\Users\dell\Desktop\Data Science\GAN\stylegan\training\dataset.py", line 71, in __init__
    assert os.path.isdir(self.tfrecord_dir)
AssertionError

途径如下：

tfrecords_path：

​

training.py和tfrecords_path：

​

Answer 1

简短回答：

你必须设置self.tfrecord_dir= '../dataset/'

长答案：

我看了一下文件。好的，基本上，错误来自行：

assert os.path.isdir(self.tfrecord_dir)

这意味着您在self.tfrecord_dir中插入的路径不被识别为目录。

你设定：self.tfrecord_dir= '\stylegan\dataset'

这是您的文件结构：

/stylegan/dataset/ 
/stylegan/training/dataset.py

您正在从training目录中执行代码，您希望在stylegan中增加两个位置，以便能够访问dataset。

在path之前添加./允许您访问当前位置的父目录的文件。如果添加../，就可以得到所需的东西。这使我们有：

self.tfrecord_dir= '../dataset/'

Answer 2

我发现answer.You必须设置self.tfrecord_dir= 'mars'，数据集应该在./datasets/mars文件夹下。