diff options
author | Soumith Chintala <soumith@gmail.com> | 2015-04-26 22:41:37 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2015-04-26 22:41:37 +0300 |
commit | 1f551bb907107d90d0598d227ac03f9d526746c3 (patch) | |
tree | 4bfeed004a1e51a5c6c61bc9dcaeec61ff12154b /init.lua | |
parent | af5084f24896d9960304a1cef1609d59edc9e733 (diff) | |
parent | 45da33035308f5fdbe85556028cb19f87c2c7519 (diff) |
Merge pull request #65 from torch/jpegperf
fixing y2jet for other default tensor types
Diffstat (limited to 'init.lua')
-rwxr-xr-x | init.lua | 9 |
1 files changed, 5 insertions, 4 deletions
@@ -1748,18 +1748,19 @@ function image.y2jet(...) -- accept 3D grayscale if input:dim() == 3 and input:size(1) == 1 then - input = torch.Tensor(input):resize(input:size(2), input:size(3)) + input = input.new(input):resize(input:size(2), input:size(3)) end -- accept 1D greyscale if input:dim() == 1 then - input = torch.Tensor(input):resize(1, input:size(1)) + input = input.new(input):resize(1, input:size(1)) end - local output = output or torch.Tensor():typeAs(input) + local output = output or input.new() local L = input:max() - input.image.colorize(output, input-1, image.jetColormap(L)) + local colorMap = image.jetColormap(L):typeAs(input) + input.image.colorize(output, input-1, colorMap) return output end |