我可以转移我的 tfrecords 并在另一台计算机上使用它们吗?

问题描述

我正在本地构建一个 tensorflow_object_detection_api 设置,希望将所有设置转移到我可以通过我的学校访问的计算机集群,并在那里训练模型。在集群上的共享 linux 系统上很难设置环境,因此我尝试在本地尽可能多地进行操作,并希望将所有内容都运送到那里并运行训练命令。我的问题是,我可以在本地生成 tfrecords 并将它们传输到集群吗?我问这个是因为我不知道这些记录是如何工作的,它们是否包含指向实际本地目录的链接?或者它们是否包含所有必要的信息?

附言我试图在集群上生成它们,但环境设置起来很棘手:tensorflow 和 opencv 安装在一个奇点中,需要调用该奇点才能使用 tf 或 opencv 运行任何脚本,但该奇点没有必要的要求运行从 csv 注释生成 tfrecords 的脚本。

我对其中的大部分内容都很陌生,因此感谢您提供任何帮助。

解决方法

是的。我试过了,它奏效了。显然,tfrecords 包含所有图像及其注释;我需要做的就是将权重转移到 Colab 并开始训练。