Advertisement

LLNet模型的数据读取

阅读量:

1. 引言

构建LLNet模型的数据输入模块

2. 程序开发

复制代码
    # Ref: LLNet: Deep Autoencoders for Low-light Image Enhancement
    #
    # Author: HSW
    # Date: 2018-05-11 
    #
    
    import tensorflow as tf
    import numpy as np
    from   PIL import Image 
    import os
    import sys
    import cv2
    import sklearn.preprocessing as prep
    
    # Training/Testing Samples 
    TRAIN_NUM_SAMPLES = 7481
    TEST_NUM_SAMPLES  = 7518
    
    class LLNet_Data(object):
    
    def __init__(self, root_dir, split="training"):
        self.root_dir = data_set_dir
        self.split        = split
        se

全部评论 (0)

还没有任何评论哟~