当我在使用MXnet开发的CNN网络中使用ROI-Pooling时,如何执行数据扩充?
例如,假设我有一个使用roi-pooling层的resnet50架构,我……
你应该能够重新利用 ImageDetRecordIter 为了这。它适用于包含边界框的对象检测数据,但您可以将边界框定义为ROI。现在,当您应用增强操作(例如翻转和旋转)时,边界框的坐标将与图像一致地进行调整。
ImageDetRecordIter
否则你可以轻松编写自己的 transform 使用Gluon的功能,可以使用任何OpenCV增强功能来应用于您的图像和ROI。只需编写一个获取数据和标签的函数,然后返回增强的数据和标签。
transform