Skip to content

※important※ Mistake in nms_boxes() of util.py #19

@YsYusaito

Description

@YsYusaito

Hi, @guichristmann

There are mistakes in NMS processing.
・errors

  1. NMS applies regardless of class.
  2. Inappropriate class reference during NMS processing

・code corrections
■utils.py - line 120,121
delete 「#」(comment out)
■utils.py - line 128,136
correct indents following bellow

※before correction
□□□if len(to_remove) != 0:
□□□□□# Remove boxes
□□□□□for r in to_remove[::-1]:
□□□□□□□del boxes[r]
□□□□□□□del scores[r]
□□□□□□□del classes[r]
□□□□□□□i += 1

※after correction
□□□if len(to_remove) != 0:
□□□□□# Remove boxes
□□□□□for r in to_remove[::-1]:
□□□□□□□del boxes[r]
□□□□□□□del scores[r]
□□□□□□□del classes[r]
□□□i += 1

By correcting like this, I could obtain appropriate inference result.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions