资源算法sketch_rnn_classification

sketch_rnn_classification

2020-02-06 | |  25 |   0 |   0

sketch_rnn_classification

This repository contains code to classify sketches from the QuickDraw dataset provided by Google - https://github.com/googlecreativelab/quickdraw-dataset

This code utilizes the same structure as sketch_rnn code provided by Google here - https://github.com/tensorflow/magenta/tree/master/magenta/models/sketch_rnn with a few modifications. This model is built by replacing the decoder with a hidden linear layer and replacing the reconstruction and KL divergence losses by cross entropy loss for classification. The code works with the magenta environment.

This code was tested on the following 16 classes from the dataset - moon, mountain, rain, sun, tree, castle, garden, house, key, teddy_bear, monkey, mermaid, duck, dog, bicycle, and angel. The training set for the experiment was created by combining the 70000 training sketches from each class. The validation set was created by combining the 2500 valodation sketches from each class. This code achieves an accuracy of 92% on these 16 classes - the confusion matrix for the validation set of these classes is shown below.


图片.png


上一篇:tensorflow-sketch-rnn-example

下一篇:sketch-rnn-poster

用户评价
全部评价

热门资源

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • shih-styletransfer

    shih-styletransfer Code from Style Transfer ...