Почему пути длины возвращают 0, используя Pytorch? - PullRequest
0 голосов
/ 30 мая 2019

Эта функция возвращает ошибку, аналогичную приведенной ниже, в то время как пути верны.Я неправильно понимаю, как работает путь?(Этот фрагмент из github-repo )

paths = paths[::skip_frame]

ValueError: slice step cannot be zero

Вот основная функция:

 def make_gif(paths, gif_path, max_frame=50, prefix=""):
      import imagei

paths.sort()
skip_frame = len(paths) // max_frame

paths = paths[::skip_frame]

images = [imageio.imread(path) for path in paths]
max_h, max_w, max_c = np.max(
        np.array([image.shape for image in images]), 0)

Вот как вызывается функция:

parser = argparse.ArgumentParser()
parser.add_argument("--model_name", type=str)
parser.add_argument("--max_frame", type=int, default=50)
parser.add_argument("--output", type=str, default="sampe.gif")
parser.add_argument("--title", type=str, default="")

if __name__ == "__main__":
    args = parser.parse_args()


    paths = glob(f"./logs/{args.model_name}/networks/*.png")

    make_gif(paths, args.output,
            max_frame=args.max_frame,
            prefix=f"{args.title}\n" if args.title else "")

1 Ответ

0 голосов
/ 31 мая 2019

Вы получаете эту ошибку, потому что ваш skip_frame равен нулю. Убедитесь, что это значение как минимум один.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...