Skip to content

Detection misclassifying due to mismatched indexing #4

Description

@kodaAtIbex

coco91_labels.txt in the repo has exactly the same classes in the same order that rf-detr outputs, however, this project doesn't seem to take into account that rf-detr skips some indices. The first skip is index 12, so every class after index 11 "fire hydrant" will be misclassified.

For example, I've observed:

  • "laptop" mislabeled as "refrigerator"
  • "keyboard" mislabeled as "vase"
  • "tv" (monitor really) mislabeled as "sink"
  • "bottle" mislabeled as "knife"
  • "mouse" mislabeled as "book"

I compared coco91_labels.txt in this repo to coco_classes.py in roboflow/rf-detr in a spreadsheet (index_mismatch_notes.ods) and it matches up with the misclassifications I've observed.

EDIT: I was able to address this for myself by putting placeholder lines in the labels file: coco91_labels.txt (patched)

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions