训练 EfficientDet D7 进行对象检测需要多少内存?

数据挖掘 Python 深度学习 张量流 物体检测
2022-02-19 13:46:03

我计划在收集图像数据库并注释图像后进行一些对象检测。我将使用 TensorFlow 2 detection Model Zoo,因为它提供了大量可供微调的预训练模型。我正在使用 TensorFlow 2.3.0 和 Python 3.8。

我将收集的图像将缩小到 1064 x 1064,以获得良好的尺寸以保持分辨率。我正在考虑使用 EfficientDet D7 1536x1536 模型;图像将被填充到最大尺寸以使算法工作。使用 1 张图像的批量大小进行训练,训练该算法需要多少内存?我知道这是一个需要很多时间的大型算法。

我可以访问以下机器:

  • 机器 1:Intel i7 - 5960x 3 GHz 8 核,GPU = Nvidia GeForce GTX TITAN X,RAM = 32 GB
  • 机器 2:Intel i7 - 5960x 3GHz 8 核,GPU = Nvidia GeForce GTX 1080,RAM = 64 GB

这些机器中的一台是否足以与此模型一起使用?此时,我计划在没有 Google Colab 的情况下进行训练并使用 GPU。

0个回答
没有发现任何回复~