我学习MXNet框架并尝试使用SSD运行对象检测示例:https://gluon.mxnet.io/chapter08_computer-vision/object-detection.html
我使用GPU是NVidia GTX 1050,4GB进行培训。一世 …
感谢您提供版本信息。你是绝对正确的 - MXNet 1.3.0中有一个错误 ImageDetIter 无限期地循环你的例子。这是固定的 2018年12月 如果您升级到MXNet 1.4.0,您将看不到问题。我通过运行上面的代码证实了这一点。
ImageDetIter
的 另一个重要的注意事项,“深度学习 - 直接涂料”已被弃用(深入学习深入学习)(d2l.ai)。内容已更新并用于MXNet课程。 强> 这是相应的 章节 在书里。
此外,还会发布课程中的视频 这里 ,如果你想看他们。
至于repro,我跑了并确认这是在1.3.x中无限循环并在1.4.0中修复。
train_iter = image.ImageDetIter( batch_size=1000, data_shape=(3, data_shape, data_shape), path_imgrec='./data/pikachu_train.rec', path_imgidx='./data/pikachu_train.idx', #shuffle=True, #mean=True, #rand_crop=1, min_object_covered=0.95, last_batch_handle='pad', max_attempts=5) train_iter.reset() for i,data in enumerate(train_iter): print((i+1)) # goes forever on 1.3.0 but not 1.4.0
希望有所帮助,
Vishaal