Searched defs:image_shape (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/ |
| D | cutmix_batch_op.cc | 67 std::vector<int64_t> image_shape = input.at(0)->shape().AsVector(); in ValidateCutMixBatch() local
107 std::vector<int64_t> image_shape = input.at(0)->shape().AsVector(); in ComputeImage() local
187 std::vector<int64_t> image_shape = input.at(0)->shape().AsVector(); in Compute() local
|
| D | mixup_batch_op.cc | 87 std::vector<int64_t> image_shape = input.at(0)->shape().AsVector(); in Compute() local
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/ |
| D | yolov3.py | 693 def construct(self, box_xy, box_wh, box_confidence, box_probs, image_shape): argument
742 def construct(self, x, image_shape): argument
|
| /third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/ |
| D | dataset.py | 297 def _infer_data(image_bgr, image_shape, gt_box_new, gt_label_new, gt_iscrowd_new_revert): argument
|