bigdataman/faces_glintasia

Download LINK https://drive.google.com/drive/folders/1IpR5mRFVFwPfWtcy-HO8Xj2M-TCZKxKK?usp=sharing

!pip list
!pip install matplotlib
!pip install --no-build-isolation scikit-learn
!pip install numpy scipy cython
!pip install mxnet
!pip install opencv-python
!pip install numpy==1.23.5
!pip install mxnet
import mxnet as mx
from mxnet import recordio
import matplotlib.pyplot as plt
import cv2
import os

path_imgidx = 'train.idx' # path to train.rec
path_imgrec = 'train.rec' # path to train.idx

imgrec = recordio.MXIndexedRecordIO(path_imgidx, path_imgrec, 'r')

i = 0
while True:
    try:
        print(i)
        header, s = recordio.unpack(imgrec.read_idx(i+1))
        #print(str(header.label))
        #img = np.array(mx.image.imdecode(s))
        img = mx.image.imdecode(s).asnumpy()
        #print(type(img))
        path = os.path.join('images',str(header.label))
        if not os.path.exists(path):
                os.makedirs(path)
        path = os.path.join(path,str(i))
        #fig = plt.figure(frameon=False)
        #fig.set_size_inches(124,124)
        #ax = plt.Axes(fig, [0., 0., 1., 1.])
        #ax.set_axis_off()
        #fig.add_axes(ax)
        #ax.imshow(img, aspect='auto')
        #dpi=1
        #fname= str(i)+'jpg'
        #fig.savefig(fname, dpi)
        #plt.savefig(path+'.jpg',bbox_inches='tight',pad_inches=0)
        (b,g,r)=cv2.split(img)
        img = cv2.merge([r,g,b])
        #w,h = img.size
        #print((img.shape))
        cv2.imwrite(path+'.jpg',img)
        i += 1
    except EOFError:
        break

# 1~ 2369931.jpg 17gb