Hi, @guichristmann
There are mistakes in NMS processing.
・errors
- NMS applies regardless of class.
- 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.
Hi, @guichristmann
There are mistakes in NMS processing.
・errors
・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.