Skip to content
Caffe Loss Layer for Ordinal Regression with Multiple Output CNN for Age Estimation.
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
layers update Jul 6, 2016
README.md
train_test.prototxt add prototxt Dec 27, 2016

README.md

Ordinal Regression

Caffe Loss Layer for Ordinal Regression with Multiple Output CNN for Age Estimation.

How to

You need to install Caffe first. Copy relative files to Caffe's source code tree.

export CAFFE_HOME=/path/to/caffe
cp layers/ordinal_regression_loss_layer.hpp $CAFFE_HOME/include/caffe/layers/ordinal_regression_loss_layer.hpp
cp layers/ordinal_regression_loss_layer.cpp $CAFFE_HOME/src/caffe/layers/ordinal_regression_loss_layer.cpp
cp layers/ordinal_regression_loss_layer.cu $CAFFE_HOME/src/caffe/layers/ordinal_regression_loss_layer.cu
cp layers/test_ordinal_regression_loss_layer.cpp $CAFFE_HOME/src/caffe/test/test_ordinal_regression_loss_layer.cpp

Modify $CAFFE_HOME/src/caffe/proto/caffe.proto according to layers/caffe.proto

After all, follow Caffe's documents and compile it.

Test the layer

make runtest GTEST_FILTER='OrdinalRegressionLossLayerTest/*'

References

You can’t perform that action at this time.