pytorch-yolo-v3
pytorch-yolo-v3 copied to clipboard
Extracting the name of the classes detected.
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.
Also I believe the colours are generated randomly,is there a way to classify based on classes?