Search code examples
tensorflowobject-detectionyolodarkflow

Is there any way to rename labels in Darkflow pretrained dataset?

I encountered a problem when running on a video. When it detects a street lamp, it shows a bird. Is there anyway to rename the label or remove the label?

I have tried to edit the misc.py but it is not working. Below is the screenshot of the detected image. enter image description here

labels20 = ["aeroplane", "bicycle", "lamp", "boat", "bottle",
    "bus", "car", "cat", "chair", "cow", "diningtable", "dog",
    "horse", "motorbike", "person", "pottedplant", "sheep", "sofa",
     "train", "tvmonitor"]

I am trying to get rid of the label or if I can, change the labels name.

Solution

  • You should change bird to lamp inside a file with extension .names, not inside the code.