TinyYolo Deeplearning4j



我已经通读了本教程 http://emaraic.com/blog/yolo-custom-object-detector,我正在使用TinyYolo和deeplearning4j,但我不太确定我是否需要更多配置来处理720p图像,因为此示例中的图像是416x416。这有什么硬性要求吗?我只是在努力完全理解一些配置,如前所述:

private static final int CHANNELS = 3;
private static final int GRID_WIDTH = 13;
private static final int GRID_HEIGHT = 13;
private static final int CLASSES_NUMBER = 1;
private static final int BOXES_NUMBER = 5;
private static final double[][] PRIOR_BOXES = {{1.5, 1.5}, {2, 2}, {3,3}, {3.5, 8}, {4, 9}};//anchors boxes
private static final int BATCH_SIZE = 4;
private static final int EPOCHS = 50;
private static final double LEARNIGN_RATE = 0.0001;
private static final int SEED = 1234;
private static final double LAMDBA_COORD = 1.0;
private static final double LAMDBA_NO_OBJECT = 0.5;

我以前在我的标签和图像中使用过 darkflow,并取得了一些巨大的成功。我想使用 deeplearning4j 与我拥有的 Java 项目进行更多集成。由于我一直在努力导入我创建的模型,这些模型仍然可以成功地与我拥有的一些 python 代码一起使用,但似乎在导出方面有一些细微差别。

如果有人能对此有所了解,我相信应该有办法处理 720p 图像。我相信可能需要调整大小。我知道暗网和暗流自己做了这个动作。另外,如果我确实执行大小调整,我的标签注释 xml 文件是否需要更改?

感谢您的任何帮助。

您需要将 720p 输入的大小调整为 tinyolo 预期输入,尺寸为 416x416

resize(rgbaMat, resizedImage, new Size(tinyyolowidth, tinyyoloheight));

参考: https://github.com/yptheangel/dl4j-android-demo/blob/master/app/src/main/java/com/yptheangel/dl4jandroid/yolo_objdetection/ObjDetection.java

最新更新