DF2Net
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
caffe_dev @ 2ea8671
framework_image
layers
model
.gitmodules
README.md

README.md

DF2Net: A Discriminative Feature Learning and Fusion Network for RGB-D Indoor Scene Classification

This is the code for the paper DF2Net: A Discriminative Feature Learning and Fusion Network for RGB-D Indoor Scene Classification(AAAI 2018).

Installation

  1. Install prerequsites for caffe. The caffe version we use can be found in our caffe_dev repo.
  2. Add the loss layers to the caffe_dev source and include path.
  3. Compile the caffe_dev Github submodule

Our Trained Models

You can download our trained model on SUN RGB-D Dataset and NYU Dataset V2 at Baidu Netdisk or Google Driver.

Our Converted HHA Images

We use the Saurabh Gupta's code to convert the depth images. We also release our converted HHA images of SUN RGB-D Dataset and NYU Depth V2 Dataset at Google Driver.