Tag: docker mxnet

在MXnet中使用im2rec来创build带有png图像的数据集

我正在尝试按照这里的例子，并创build自己的数据集使用MXnet的培训。 我的数据按照示例中的规定进行组织： /data yes/ file1.png file2.png … no/ file1.png file2.png … 本教程说明了第一步是运行im2rec.py来创build.lst文件，然后再次运行im2rec.py （不同的选项）来创build.rec文件。 要创build.lst文件，我input： > python tools/im2rec.py my_data /data –list True –recursive True –train-ratio .75 –exts .png 完成后，将创build两个文件（按预期）， my_data_train.lst和my_data_val.lst 。 这两个文件中的总行数与我的yes/和no/目录中的文件总数相同。 然后，我尝试再次运行im2rec以使用以下命令创build.rec文件： > python tools/im2rec.py my_data /data –resize 227 –num-thread 16 这会运行几秒钟，然后（默默）崩溃。 在这个过程中，它创build了4个空文件： my_data_train.idx ， my_data_train.rec ， my_data_val.idx和my_data_val.rec 。 问题：为了能够创build一个包含我自己的.png图像的正确的.rec文件，我需要做些什么改变？ 额外详情： 我正在docker hub上由dmlc提供的docker容器（mxnet / python：gpu）中工作; […]