staged.

Chunk
1 parent b797194f
Showing 1 changed file with 21 additions and 7 deletions Show diff stats
mdata/crop.py
@@ -5,6 +5,10 @@ from PIL import Image
 from ..common import *
 import random
  
+import cv2
+import numpy as np
+
+
 base_dir = '/data/hadoop/ImageNet/ILSVRC/ILSVRC2013_DET_val/'
 category = 'Test'
  
@@ -14,14 +18,24 @@ def crop_Test():
         for name in files:
             image = os.path.join(path, name)
             print image
+
+            # try:
+            #     im = Image.open(image)
+            #     w, h = im.size
+            #     if w < 300 or h < 300:
+            #         continue
+            #     left, upper = random.randint(0, w - 300), random.randint(0, h - 300)
+            #     im = im.crop((left, upper, left + 300, upper + 300))
+            #     im.save(os.path.join(base_dir, category + '_crop_pil', name))
+            # except:
+            #     pass
+
             try:
-                im = Image.open(image)
-                w, h = im.size
-                if w < 300 or h < 300:
-                    continue
-                left, upper = random.randint(0, w - 300), random.randint(0, h - 300)
-                im = im.crop((left, upper, left + 300, upper + 300))
-                im.save(os.path.join(base_dir, category + '_crop', name))
+                img = cv2.imread(image, cv2.CV_LOAD_IMAGE_UNCHANGED)
+                h, w = img.shape[:2]
+                left, upper = np.random.randint(w - 300), np.random.randint(h - 300)
+                img_crop = img[upper:upper + 300, left:left + 300]
+                cv2.imwrite(os.path.join(base_dir, category + '_crop_cv', name))
             except:
                 pass
...	...	@@ -5,6 +5,10 @@ from PIL import Image
5	5	from ..common import *
6	6	import random
7	7
	8	+import cv2
	9	+import numpy as np
	10	+
	11	+
8	12	base_dir = '/data/hadoop/ImageNet/ILSVRC/ILSVRC2013_DET_val/'
9	13	category = 'Test'
10	14
...	...	@@ -14,14 +18,24 @@ def crop_Test():
14	18	for name in files:
15	19	image = os.path.join(path, name)
16	20	print image
	21	+
	22	+ # try:
	23	+ # im = Image.open(image)
	24	+ # w, h = im.size
	25	+ # if w < 300 or h < 300:
	26	+ # continue
	27	+ # left, upper = random.randint(0, w - 300), random.randint(0, h - 300)
	28	+ # im = im.crop((left, upper, left + 300, upper + 300))
	29	+ # im.save(os.path.join(base_dir, category + '_crop_pil', name))
	30	+ # except:
	31	+ # pass
	32	+
17	33	try:
18		- im = Image.open(image)
19		- w, h = im.size
20		- if w < 300 or h < 300:
21		- continue
22		- left, upper = random.randint(0, w - 300), random.randint(0, h - 300)
23		- im = im.crop((left, upper, left + 300, upper + 300))
24		- im.save(os.path.join(base_dir, category + '_crop', name))
	34	+ img = cv2.imread(image, cv2.CV_LOAD_IMAGE_UNCHANGED)
	35	+ h, w = img.shape[:2]
	36	+ left, upper = np.random.randint(w - 300), np.random.randint(h - 300)
	37	+ img_crop = img[upper:upper + 300, left:left + 300]
	38	+ cv2.imwrite(os.path.join(base_dir, category + '_crop_cv', name))
25	39	except:
26	40	pass
27	41
...	...