在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)中工作; 他们也在他们的github页面上提供了这个例子。 数据通过容器中的共享目录可用。 所以这可能是一个docker问题。 是什么让我有点担心这是一个docker问题是,我不得不im2rec
pip install opencv-python
为了im2rec
能够导入cv2 …我本来希望提供容器的人会照顾这个。
你是对的,图像是缺lessPython的opencv。 而不是通过点安装,请做apt-get install python-opencv
。
公关张贴在这里: 在MXnet中使用im2rec与png图像创build数据集