aisingapore / PeekingDuck

A modular framework built to simplify Computer Vision inference workloads.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

yolact_edge - need to pass detect = [ ] as integer not string class labels

tappyness1 opened this issue · comments

For Object Detection, to pass the .Node, we use detect = ["class_name0", "class_name1", ..., "class_nameN"]. However, for yolact, we will need to use detect = [0, 1, ..., N].

yolact_edge:
image

yolov4-tiny:
image

[Request] To align on which to use - integer index or class name as string

Hi, the latest release of PeekingDuck is able to accept class names for yolact_edge:
image
As can be seen above, the person got converted internally to id 0.
Please update your PeekingDuck to the latest version.