|
|
|
@@ -58,7 +58,8 @@ class ImageGradients(Cell): |
|
|
|
super(ImageGradients, self).__init__() |
|
|
|
|
|
|
|
def construct(self, images): |
|
|
|
_check_input_4d(F.shape(images), "images", self.cls_name) |
|
|
|
check = _check_input_4d(F.shape(images), "images", self.cls_name) |
|
|
|
images = F.depend(images, check) |
|
|
|
batch_size, depth, height, width = P.Shape()(images) |
|
|
|
dy = images[:, :, 1:, :] - images[:, :, :height - 1, :] |
|
|
|
dy_last = P.Fill()(P.DType()(images), (batch_size, depth, 1, width), 0) |
|
|
|
|