دریافت تصاویر از دیتاست MNIST - هفت خط کد انجمن پرسش و پاسخ برنامه نویسی

وبـــلاگ هــفت خــط کــد


آموزش های برنامه نویسی
۱۴۰ نفر آنلاین
۱ عضو و ۱۳۹ مهمان در سایت حاضرند

دریافت تصاویر از دیتاست MNIST

0 امتیاز
19 بازدید
سلام.

قصد دارم یک سری از کارها  را بر روی تصاویر مربوط به دیتاست MNIST انجام بدم همانطور که اطلاع دارید این دیتاست به صورت فایل باینری وجود داره می خواستم اگر دوستان اسکریپتی برای دیکود کردن این فایل باینری دارند در اختیار من قرار بدن. ممنون
سوال شده آذر 18, 1396  بوسیله ی ثریا (امتیاز 27)   2

1 پاسخ

0 امتیاز

16 بایت اول در فایل باینری تصویری مربوط به اطلاعات هدر فایل است و مابقی استریم داده ای اطلاعات پیسکل های تصاویر است تصاویر MNIST به صورت 28*28 است 784 بایت برای هر تصویر.

اطلاعات Labels هم به صورت 8 بایت اول مربوط به هدر و یک بایت برای هر label است.

def extract_labels(mnist_label_file_path, label_file_path):
    with open(mnist_label_file_path, "rb") as mnist_label_file:
        # 32 bit integer magic number
        mnist_label_file.read(4)
        # 32 bit integer number of items
        mnist_label_file.read(4)
        # actual test label
        label_file = open(label_file_path, "w")
        label = mnist_label_file.read(1)
        while label:
            label_file.writelines(str(label[0]) + "\n")
            label = mnist_label_file.read(1)
        label_file.close()

def extract_images(images_file_path, images_save_folder):
        # images_file_path = "./t10k-images-idx3-ubyte"
        with open(images_file_path, "rb") as images_file:
            # 32 bit integer magic number
            images_file.read(4)
            # 32 bit integer number of images
            images_file.read(4)
            # 32 bit number of rows
            images_file.read(4)
            # 32 bit number of columns
            images_file.read(4)
            # every image contain 28 x 28 = 784 byte, so read 784 bytes each time
            count = 1
            image = np.zeros((28, 28, 1), np.uint8)
            image_bytes = images_file.read(784)
            while image_bytes:
                image_unsigned_char = struct.unpack("=784B", image_bytes)
                for i in range(784):
                    image.itemset(i, image_unsigned_char[i])
                image_save_path = r"%s\%d.png" % (images_save_folder, count)

                cv2.imwrite(image_save_path, image)
                print(count)
                image_bytes = images_file.read(784)
                count += 1

 

پاسخ داده شده آذر 18, 1396 بوسیله ی LastUzer (امتیاز 40)  
...