Home
last modified time | relevance | path

Searched refs:img_shape (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/
Ddataset.py161 def rescale_column(img, img_shape, gt_bboxes, gt_label, gt_num, config): argument
179 img_shape = (config.img_height, config.img_width, 1.0)
180 img_shape = np.asarray(img_shape, dtype=np.float32)
182 return (pad_img_data, img_shape, gt_bboxes, gt_label, gt_num)
184 def rescale_column_test(img, img_shape, gt_bboxes, gt_label, gt_num, config): argument
198 img_shape = np.append(img_shape, (scale_factor, scale_factor))
199 img_shape = np.asarray(img_shape, dtype=np.float32)
201 return (pad_img_data, img_shape, gt_bboxes, gt_label, gt_num)
204 def resize_column(img, img_shape, gt_bboxes, gt_label, gt_num, config): argument
211 img_shape = (config.img_height, config.img_width, 1.0)
[all …]
Dnetwork_define.py101 def construct(self, x, img_shape, gt_bboxe, gt_label, gt_num): argument
102 …loss1, loss2, loss3, loss4, loss5, loss6 = self._backbone(x, img_shape, gt_bboxe, gt_label, gt_num)
144 def construct(self, x, img_shape, gt_bboxe, gt_label, gt_num): argument
146 loss = self.network(x, img_shape, gt_bboxe, gt_label, gt_num)
147 … grads = self.grad(self.network, weights)(x, img_shape, gt_bboxe, gt_label, gt_num, self.sens)
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/cl/
Dactivation.cl20 …el void LeakyRelu(__read_only image2d_t input, __write_only image2d_t output, const int2 img_shape,
24 if (X >= img_shape.x || Y >= img_shape.y) return;
35 __kernel void Relu(__read_only image2d_t input, __write_only image2d_t output, const int2 img_shape
38 if (X >= img_shape.x || Y >= img_shape.y) return;
44 …rnel void Relu6(__read_only image2d_t input, __write_only image2d_t output, const int2 img_shape) {
47 if (X >= img_shape.x || Y >= img_shape.y) return;
53 …oid(__read_only image2d_t input, __write_only image2d_t output, const int2 img_shape, const int c4,
57 if (X >= img_shape.x || Y >= img_shape.y || c4 == 0) return;
77 __kernel void Tanh(__read_only image2d_t input, __write_only image2d_t output, const int2 img_shape
80 if (X >= img_shape.x || Y >= img_shape.y) return;
[all …]
/third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/
Dconfig.py26 img_shape = [352, 640] variable in ConfigYOLOV3ResNet18
Dyolov3.py416 self.input_shape = Tensor(tuple(config.img_shape[::-1]), ms.float32)
519 self.input_shape = Tensor(tuple(config.img_shape[::-1]), ms.float32)
690 self.input_shape = Tensor(np.array(config.img_shape), ms.float32)
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/
Dimage_process.h58 std::vector<size_t> img_shape;
Dimage_process.cc1016 float num = static_cast<float>(config.img_shape[0]); in GetDefaultBoxes()
/third_party/mindspore/tests/ut/cpp/dataset/
Dimage_process_test.cc562 config.img_shape = {300, 300}; in TEST_F()