pytorch-yolo-v3 icon indicating copy to clipboard operation
pytorch-yolo-v3 copied to clipboard

Extracting the name of the classes detected.

Open iqrammm opened this issue 5 years ago • 1 comments

I have 3 class(person,sofa and cats). I would like to print output when cats are being detected and no output when person and sofa being detected. Been going through the code but cant seem to find ways to modify it.

iqrammm avatar Aug 04 '19 14:08 iqrammm

Also I believe the colours are generated randomly,is there a way to classify based on classes?

iqrammm avatar Aug 04 '19 14:08 iqrammm