Parcourir la source

Add -invert_x option for user_method

nagadomi il y a 8 ans
Parent
commit
cba96e36fb
2 fichiers modifiés avec 6 ajouts et 0 suppressions
  1. 4 0
      convert_data.lua
  2. 2 0
      lib/settings.lua

+ 4 - 0
convert_data.lua

@@ -118,6 +118,10 @@ local function load_images(list)
 	       -- method == user
 	       local yy = im
 	       local xx, meta2 = image_loader.load_byte(csv_meta.x)
+	       if settings.invert_x then
+		  xx = (-(xx:long()) + 255):byte()
+	       end
+
 	       if xx then
 		  if meta2 and meta2.alpha then
 		     xx = alpha_util.fill(xx, meta2.alpha, alpha_color)

+ 2 - 0
lib/settings.lua

@@ -80,6 +80,7 @@ cmd:option("-padding", 0, 'replication padding size')
 cmd:option("-padding_y_zero", 0, 'zero padding y for segmentation (0|1)')
 cmd:option("-grayscale", 0, 'grayscale x&y (0|1)')
 cmd:option("-validation_filename_split", 0, 'make validation-set based on filename(basename)')
+cmd:option("-invert_x", 0, 'invert x image in convert_lua')
 
 local function to_bool(settings, name)
    if settings[name] == 1 then
@@ -101,6 +102,7 @@ to_bool(settings, "pairwise_flip")
 to_bool(settings, "padding_y_zero")
 to_bool(settings, "grayscale")
 to_bool(settings, "validation_filename_split")
+to_bool(settings, "invert_x")
 
 if settings.plot then
    require 'gnuplot'