Searched refs:max_boxes (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | transforms.py | 136 max_boxes, label_smooth, label_smooth_factor=0.1): argument
196 pad_gt_box0 = np.zeros(shape=[max_boxes, 4], dtype=np.float32)
197 pad_gt_box1 = np.zeros(shape=[max_boxes, 4], dtype=np.float32)
198 pad_gt_box2 = np.zeros(shape=[max_boxes, 4], dtype=np.float32)
370 def _data_aug(image, box, jitter, hue, sat, val, image_input_size, max_boxes, argument
388 if len(box) > max_boxes:
389 box = box[:max_boxes]
391 box_data = np.zeros((max_boxes, 5))
437 max_boxes = config.max_box
444 image_input_size=input_size, max_boxes=max_boxes,
[all …]
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/ |
| D | dataset.py | 38 max_boxes = 20
153 box_data = np.zeros((max_boxes, 5))
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/ |
| D | image_process.h | 172 int max_boxes);
|
| D | image_process.cc | 1083 int max_boxes) { in ApplyNms() argument
1101 if (keep.size() >= max_boxes) { in ApplyNms()
|