在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.lstmy_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.idxmy_data_train.recmy_data_val.idxmy_data_val.rec

问题:为了能够创build一个包含我自己的.png图像的正确的.rec文件,我需要做些什么改变?

额外详情:

我正在docker hub上由dmlc提供的docker容器(mxnet / python:gpu)中工作; 他们也在他们的github页面上提供了这个例子。 数据通过容器中的共享目录可用。 所以这可能是一个docker问题。 是什么让我有点担心这是一个docker问题是,我不得不im2rec pip install opencv-python为了im2rec能够导入cv2 …我本来希望提供容器的人会照顾这个。

你是对的,图像是缺lessPython的opencv。 而不是通过点安装,请做apt-get install python-opencv

公关张贴在这里: 在MXnet中使用im2rec与png图像创build数据集

Interesting Posts