我计划在收集图像数据库并注释图像后进行一些对象检测。我将使用 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。