How Do I convert or running Torch/LUA code in MATLAB?
1 visualización (últimos 30 días)
Mostrar comentarios más antiguos
How Do I convert or running Torch/LUA code in MATLAB? Thank You.
colorize.lua
https://github.com/satoshiiizuka/siggraph2016_colorization
----------------------------------------------------------------------------
require 'nn' require 'nngraph' require 'image'
local infile = arg[1] local outfile = arg[2] or 'out.png' local netfile = arg[3] or 'colornet.t7'
local d = torch.load( netfile ) local datamean = d.mean local model = d.model:float()
local function pred2rgb( x, data ) local I = torch.cat( data[1][{{1},{},{}}]:float(), data[1]:clone():float():mul(2):add(-1), 1) local O = image.scale( I, x:size(3), x:size(2) ) local X = image.rgb2lab( image.yuv2rgb( torch.repeatTensor( x, 3, 1, 1 ) ) ) O = O*100 O[1] = X[1] O = image.rgb2yuv( image.lab2rgb( O ) ) return image.yuv2rgb( torch.cat( x, O[{{2,3},{},{}}], 1 ) ) end
local I = image.load( infile ) if I:size(1)==3 then I = image.rgb2y(I) end local X2 = image.scale( I, torch.round(I:size(3)/8)*8, torch.round(I:size(2)/8)*8 ):add(-datamean):float() local X1 = image.scale( X2, 224, 224 ):float() X1 = X1:reshape( 1, X1:size(1), X1:size(2), X1:size(3) ) X2 = X2:reshape( 1, X2:size(1), X2:size(2), X2:size(3) ) model.forwardnodes[9].data.module.modules[3].nfeatures = X2:size(3)/8 model.forwardnodes[9].data.module.modules[4].nfeatures = X2:size(4)/8
image.save( outfile, pred2rgb( I:float(), model:forward( {X1, X2} ) ) )
0 comentarios
Respuestas (0)
Ver también
Categorías
Más información sobre Logical en Help Center y File Exchange.
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!