Extract images from recordio format

extract-img-recordio.py

from mxnet import recordio
import mxnet as mx
import cv2
import matplotlib.pyplot as plt
from PIL import Image

record = mx.recordio.MXRecordIO('mlclassify_train.rec', 'r')

i = 0
while True:
    item = record.read()
    if not item:
        break
    header, img = mx.recordio.unpack_img(item)
    filename = "aeroplane%s.jpg" % i
    cv2.imwrite(filename, img)
    i += 1
    b,g,r = cv2.split(img)
    img = cv2.merge([r,g,b])
    plt.imshow(img)
record.close()