All classes probabilities for particular image

[Venky0892]

Hi, is it possible to get all the class probabilities in the output of an image just to ensure how good the model is capable of detecting an object ?

[glenn-jocher]

@Venky0892 no. This information is only available at the NMS input:

yolov5/utils/general.py

Lines 699 to 706 in 055e72a

	def non_max_suppression(prediction, conf_thres=0.25, iou_thres=0.45, classes=None, agnostic=False, multi_label=False,
	labels=(), max_det=300):
	"""Runs Non-Maximum Suppression (NMS) on inference results
	Returns:
	list of detections, on (n,6) tensor per image [xyxy, conf, cls]
	"""