Skip to content

Instantly share code, notes, and snippets.

View dersmon's full-sized avatar

Simon Hohl dersmon

  • Deutsches Archäologisches Institut
  • Berlin, Germany
View GitHub Profile
@dersmon
dersmon / prediction.py
Last active May 31, 2016 09:13
Using Transfomer for prediction..
import sys
import caffe
import cv2
import Image
import numpy as np
from scipy.misc import imresize
caffe_root = "/home/simon/Workspaces/caffe/"
#MODEL_FILE = caffe_root + 'models/placesCNN/places205CNN_deploy.prototxt'
@dersmon
dersmon / prediction.py
Created November 12, 2015 10:23
Using trained caffe model in python script, added value scaling and mean.
import sys
import caffe
import cv2
import Image
import numpy as np
from scipy.misc import imresize
caffe_root = "/home/simon/Workspaces/caffe/"
#MODEL_FILE = caffe_root + 'models/placesCNN/places205CNN_deploy.prototxt'
@dersmon
dersmon / deploy.prototxt
Created November 11, 2015 13:11
Training und deployment prototxt Files for customized AlexNet
name: "AlexNet"
input: "data"
input_shape {
dim: 10
dim: 3
dim: 227
dim: 227
}
layer {
name: "conv1"
@dersmon
dersmon / prediction.py
Created November 11, 2015 13:03
Using a trained caffe model (buggy).
import sys
import caffe
import cv2
import Image
import numpy as np
from scipy.misc import imresize
caffe_root = "/home/simon/caffe/"
#MODEL_FILE = caffe_root + 'models/placesCNN/places205CNN_deploy.prototxt'
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<groupId>net.dersmon</groupId>
<artifactId>neural-network-training</artifactId>
<version>0.0.1-SNAPSHOT</version>
<name>neural-network-training</name>
<url>http://maven.apache.org</url>
<packaging>jar</packaging>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<groupId>net.dersmon</groupId>
<artifactId>dl4j-imagepipeline-tutorial</artifactId>
<version>0.0.1-SNAPSHOT</version>
<name>dl4j-imagepipeline-tutorial</name>
<url>http://maven.apache.org</url>
<packaging>jar</packaging>
import org.canova.api.records.reader.RecordReader;
import org.canova.api.split.FileSplit;
import org.canova.image.recordreader.ImageRecordReader;
import org.deeplearning4j.datasets.canova.RecordReaderDataSetIterator;
import org.deeplearning4j.datasets.fetchers.LFWDataFetcher;
import org.deeplearning4j.datasets.iterator.DataSetIterator;
import org.deeplearning4j.datasets.iterator.impl.LFWDataSetIterator;
import org.deeplearning4j.eval.Evaluation;
import org.deeplearning4j.nn.api.OptimizationAlgorithm;
import org.deeplearning4j.nn.conf.MultiLayerConfiguration;
PROTOC src/caffe/proto/caffe.proto
CXX .build_release/src/caffe/proto/caffe.pb.cc
CXX src/caffe/parallel.cpp
CXX src/caffe/solver.cpp
CXX src/caffe/layer.cpp
CXX src/caffe/util/signal_handler.cpp
CXX src/caffe/util/db_lmdb.cpp
CXX src/caffe/util/blocking_queue.cpp
CXX src/caffe/util/io.cpp
CXX src/caffe/util/math_functions.cpp
// this
String labeledPath = System.getProperty("user.home") + "/Pictures/labeled_faces_in_the_wild";
List<String> labels = new ArrayList<>();
for (File f : new File(labeledPath).listFiles()) {
labels.add(f.getName());
}
RecordReader recordReader = new ImageRecordReader(28, 28, true, labels);
recordReader.initialize(new FileSplit(new File(labeledPath)));