Merge NMS update · a322fc5d4b - greenhouse - Gitea: Git with a cup of tea

apoorva/greenhouse

Merge NMS update

This commit is contained in:

Glenn Jocher

2020-03-26 12:48:00 -07:00

parent 171b4129b5

commit a322fc5d4b

1 changed files with 1 additions and 0 deletions

									
										utils/utils.py
									
		+1
		
												View File
												
					@@ -564,6 +564,7 @@ def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, multi_label=T

					            i = iou.max(0)[0] < iou_thres

					            i = iou.max(0)[0] < iou_thres

					        output[xi] = x[i]

					        output[xi] = x[i]

					    return output

					def get_yolo_layers(model):

					def get_yolo_layers(model):