提交 eaa12937 authored 作者: João Victor Risso's avatar João Victor Risso

Add remark on the format of the input tensor in dnn_spatialtf

上级 a6165acc
...@@ -2974,7 +2974,10 @@ def dnn_spatialtf(img, theta, scale_width=1, scale_height=1, precision=theano.co ...@@ -2974,7 +2974,10 @@ def dnn_spatialtf(img, theta, scale_width=1, scale_height=1, precision=theano.co
Parameters Parameters
---------- ----------
img : tensor img : tensor
Images to which the transformations will be applied. Images to which the transformations will be applied. The implementation
assumes the tensor is in NCHW format, where N is the number of images,
C is the number of color channels, H is the height of the inputs, and
W is width of the inputs.
theta : tensor theta : tensor
Affine transformation tensor containing one affine transformation Affine transformation tensor containing one affine transformation
matrix per image. ``theta`` is usually generated by the localization matrix per image. ``theta`` is usually generated by the localization
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论