#11 and filename bug fix

Author: ildoonet

RandAugment/augmentations.py

@@ -90,7 +90,7 @@ def SolarizeAdd(img, addition=0, threshold=128):
 
 
 def Posterize(img, v):  # [4, 8]
-    v = int(v)
+    v = int(v) - 1
     return PIL.ImageOps.posterize(img, v)
 
 

RandAugment/train.py

@@ -258,7 +258,7 @@ def train_and_eval(tag, dataroot, test_ratio=0.0, cv_fold=0, reporter=None, metr
             logger.warning('Provide --save argument to save the checkpoint. Without it, training result will not be saved!')
 
     if args.save:
-        add_filehandler(logger, args.save.replace('.pth', '.log'))
+        add_filehandler(logger, args.save.replace('.pth', '') + '.log')
 
     logger.info(json.dumps(C.get().conf, indent=4))