public class CifarLoader extends BaseImageLoader implements Serializable
| Modifier and Type | Field and Description |
|---|---|
static int |
BYTEFILELEN |
static int |
CHANNELS |
static Map<String,String> |
cifarTrainData |
static String |
dataBinFile |
static String |
dataBinUrl |
String |
dataFile |
String |
dataUrl |
protected static File |
fullDir |
static int |
HEIGHT |
protected static String |
labelFileName |
static File |
LABELPATH |
protected List<String> |
labels |
static String |
localDir |
static int |
NUM_LABELS |
static int |
NUM_TEST_IMAGES |
static int |
NUM_TRAIN_IMAGES |
protected int |
numExamples |
protected int |
numLabels |
protected String |
regexPattern |
protected static String |
testFileName |
static File |
TESTPATH |
protected static String[] |
trainFileNames |
static File |
TRAINPATH |
protected String |
version |
static int |
WIDTH |
ALLOWED_FORMATS, BASE_DIR, centerCropIfNeeded, channels, height, log, rng, width| Constructor and Description |
|---|
CifarLoader() |
CifarLoader(String version) |
CifarLoader(String version,
String localDir) |
| Modifier and Type | Method and Description |
|---|---|
org.nd4j.linalg.api.ndarray.INDArray |
asMatrix(File f) |
org.nd4j.linalg.api.ndarray.INDArray |
asMatrix(InputStream inputStream) |
org.nd4j.linalg.api.ndarray.INDArray |
asRowVector(File f) |
org.nd4j.linalg.api.ndarray.INDArray |
asRowVector(InputStream inputStream) |
void |
generateMaps() |
InputStream |
getInputStream() |
List<String> |
getLabels() |
boolean |
imageFilesExist() |
void |
load() |
downloadAndUntar, getAllowedFormatspublic static final int NUM_TRAIN_IMAGES
public static final int NUM_TEST_IMAGES
public static final int NUM_LABELS
public static final int HEIGHT
public static final int WIDTH
public static final int CHANNELS
public static final int BYTEFILELEN
public String dataUrl
public String dataFile
public static String dataBinUrl
public static String dataBinFile
protected static String labelFileName
protected static String[] trainFileNames
protected static String testFileName
public static String localDir
protected static File fullDir
protected String regexPattern
protected int numExamples
protected int numLabels
protected String version
public static final File TRAINPATH
public static final File TESTPATH
public static final File LABELPATH
public CifarLoader(String version)
public CifarLoader()
public org.nd4j.linalg.api.ndarray.INDArray asRowVector(File f) throws IOException
asRowVector in class BaseImageLoaderIOExceptionpublic org.nd4j.linalg.api.ndarray.INDArray asRowVector(InputStream inputStream) throws IOException
asRowVector in class BaseImageLoaderIOExceptionpublic org.nd4j.linalg.api.ndarray.INDArray asMatrix(File f) throws IOException
asMatrix in class BaseImageLoaderIOExceptionpublic org.nd4j.linalg.api.ndarray.INDArray asMatrix(InputStream inputStream) throws IOException
asMatrix in class BaseImageLoaderIOExceptionpublic void generateMaps()
public void load()
public boolean imageFilesExist()
public InputStream getInputStream()
Copyright © 2016. All rights reserved.