Increase jpeg corruption & add error

2020-10-28 17:37:39 -06:00 · 2020-10-28 17:37:39 -06:00 · 25b007a0f5
commit 25b007a0f5
parent 796659b0ac
1 changed files with 4 additions and 2 deletions
--- a/codes/data/image_corruptor.py
+++ b/codes/data/image_corruptor.py
@ -101,8 +101,10 @@ class ImageCorruptor:
                    lo=15
                    range=60
                elif aug == 'jpeg-normal':
-                    lo=35
-                    range=60
+                    lo=55
+                    range=35
+                else:
+                    raise NotImplementedError("specified jpeg corruption doesn't exist")
                # JPEG compression
                qf = (rand_int % range + lo)
                # cv2's jpeg compression is "odd". It introduces artifacts. Use PIL instead.