为什么长度路径使用Pytorch返回0？

Question

当路径正确时，此函数返回与以下相同的错误。我是否误解了路径的工作方式？ （此摘录来自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 "")

Answer 1

由于您的skip_frame为零，您会收到此错误。确保其值至少为一个。