/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/ |
D | dataset.py | 161 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 …]
|
D | network_define.py | 101 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/ |
D | activation.cl | 20 …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/ |
D | config.py | 26 img_shape = [352, 640] variable in ConfigYOLOV3ResNet18
|
D | yolov3.py | 416 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/ |
D | image_process.h | 58 std::vector<size_t> img_shape;
|
D | image_process.cc | 1016 float num = static_cast<float>(config.img_shape[0]); in GetDefaultBoxes()
|
/third_party/mindspore/tests/ut/cpp/dataset/ |
D | image_process_test.cc | 562 config.img_shape = {300, 300}; in TEST_F()
|