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)
还没有任何评论哟~
