Skip to content

Files

Latest commit

 

History

History

illustration2vec

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Sep 23, 2020
Oct 29, 2020
Apr 11, 2023
Sep 23, 2020
Sep 23, 2020

Illustration2Vec

input

input_image
Hatsune Miku (初音ミク), © Crypton Future Media, INC., http://piapro.net/en_for_creators.html. This image is licensed under the Creative Commons - Attribution-NonCommercial, 3.0 Unported (CC BY-NC).

  • Ailia input shape: (1, 3, 224, 224)

output

  • Estimating Tag Mode
[{'character': [('hatsune miku', 0.9999994039535522)],
  'copyright': [('vocaloid', 0.9999998807907104)],
  'general': [('thighhighs', 0.9957002997398376),
              ('1girl', 0.9873842000961304),
              ('twintails', 0.9814184308052063),
              ('solo', 0.9636198878288269),
              ('aqua hair', 0.9165698885917664),
              ('long hair', 0.8823321461677551),
              ('very long hair', 0.8341774940490723),
              ('detached sleeves', 0.7452000975608826),
              ('skirt', 0.6787945628166199),
              ('necktie', 0.5615589618682861),
              ('aqua eyes', 0.552979588508606)],
  'rating': [('safe', 0.9787901043891907),
             ('questionable', 0.020344465970993042),
             ('explicit', 0.0006209250423125923)]}]
  • Extracting Feature Vector Mode
[[ 7.4673967   3.6940672   0.53971916 ... -0.08819806  2.720264
   7.3380337 ]]

usage

Automatically downloads the onnx and prototxt files on the first run. It is necessary to be connected to the Internet while downloading.

For the sample image,

$ python3 illustration2vec.py

If you want to specify the input image, put the image path after the --input option.

$ python3 illustration2vec.py --input IMAGE_PATH

By adding the --video option, you can input the video.
If you pass 0 as an argument to VIDEO_PATH, you can use the webcam input instead of the video file.

$ python3 illustration2vec.py --video VIDEO_PATH

The default mode is to predict the tag, but you can switch to the mode to extract the feature vector by specifying --featurevec.

$ python3 illustration2vec --featurevec

Reference

Framework

Caffe

Model Format

ONNX opset = 10

Netron