search

ShusenTang / Dive-into-DL-PyTorch

本项目将《动手学深度学习》(Dive into Deep Learning)原书中的MXNet实现改为PyTorch实现。
http://tangshusen.me/Dive-into-DL-PyTorch
Apache License 2.0
18.17k stars 5.38k forks source link

non_max_suppression 函数 死循环 #91

Open wmpscc opened 4 years ago

wmpscc commented 4 years ago

bug描述

9.4节代码的non_max_suppression函数，存在死循环。

while len(sorted_bb_info_list) != 0:

这个条件可以无法退出。

版本信息 pytorch:1.3.1 torchvision:0.4.2 torchtext:0.4.0 ...