From 180e7676e4da1fc18fbf9c99edd177f86cab50a2 Mon Sep 17 00:00:00 2001 From: nagadomi Date: Sun, 8 Nov 2015 06:31:29 +0900 Subject: [PATCH] Drop full transparent background --- lib/image_loader.lua | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/lib/image_loader.lua b/lib/image_loader.lua index 94dd95a..dfb3167 100644 --- a/lib/image_loader.lua +++ b/lib/image_loader.lua @@ -6,6 +6,7 @@ local image_loader = {} local clip_eta8 = (1.0 / 255.0) * 0.5 - (1.0e-7 * (1.0 / 255.0) * 0.5) local clip_eta16 = (1.0 / 65535.0) * 0.5 - (1.0e-7 * (1.0 / 65535.0) * 0.5) +local background_color = 0.5 function image_loader.decode_float(blob) local im, alpha = image_loader.decode_byte(blob) @@ -87,6 +88,11 @@ function image_loader.decode_byte(blob) local sum_alpha = (im[4] - 1.0):sum() if sum_alpha < 0 then alpha = im[4]:reshape(1, im:size(2), im:size(3)) + -- drop full transparent background + local mask = torch.le(alpha, 0.0) + im[1][mask] = background_color + im[2][mask] = background_color + im[3][mask] = background_color end local new_im = torch.FloatTensor(3, im:size(2), im:size(3)) new_im[1]:copy(im[1])