Home

Awesome

pytorch-multi-label-classifier

Introdution

A pytorch implemented classifier for Multiple-Label classification. You can easily train, test your multi-label classification model and visualize the training process.
Below is an example visualizing the training of one-label classifier. If you have more than one attributes, no doubt that all the loss and accuracy curves of each attribute will show on web browser orderly.

LossAccuracy

Module

Multi-Label Data Format

Data Format Explanation.

 Store attribute information including its name and value. label.txt example. Lines in label.txt stack as follows:

 Note: mind the difference between attribute and attribute value.

 Store objects information including attribute id and bounding box and so on. Each line is one json dict recording one object. data.txt example  

Dependence

TODO

Reference

Part of codes and models refer to some other OSS listed below for thanks: