Git Product home page Git Product logo

joint-learning-of-nn's Introduction

Joint-Learning-of-NN

The full paper can be found in here: https://arxiv.org/abs/1905.06526

Results on Generative Model Data Social Network

Requirements

  • Python 2.7
  • TensorFlow >= 1.2rc0
  • Numpy
  • Tensorflow slim library [https://github.com/tensorflow/models/tree/master/research/slim] (for inception v3 architecture, you can use custom built network architecture if you have one. Since the five datasets shown in the paper have different number of classes, I added 4 more last layers for inception v3. Please make changes based on your need.)

Content

  • images.py: Script to run the training process.
  • utils.py: Utility functions.
  • datagenerator.py: Contains a wrapper class for the new input pipeline.
  • images/*: contains some teaser images from the paper.

Usage

All you need to touch is the images.py. You can configure different parameters in there. You have to provide .txt files to the script (exp_1_train.txt, exp_2_train.txt, .... and exp_1_test.txt, exp_2_test.txt, .... for different datasets. In the paper, I used five datasets.) Each of them list the complete path to your train/val images together with the class number in the following structure.

Example train.txt:
/path/to/train/image1.png 0
/path/to/train/image2.png 1
/path/to/train/image3.png 2
/path/to/train/image4.png 0
.
.

were the first column is the path and the second the class label.

In the paper and in the current training script, I used five datasets:

  1. Caltech-UCSD Birds 200
  2. Stanford Dogs Dataset
  3. Flower Datasets
  4. Cars Dataset
  5. FGVC-Aircraft Benchmark

joint-learning-of-nn's People

Contributors

zaiweizhang avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.