From 48cd6ed71771db6a09d8a919d29fe1caa4203b5a Mon Sep 17 00:00:00 2001 From: Chunk Date: Sun, 19 Apr 2015 17:50:27 +0800 Subject: [PATCH] staged. --- mdata/crop.py | 28 +++++++++++++++++++++------- 1 file changed, 21 insertions(+), 7 deletions(-) diff --git a/mdata/crop.py b/mdata/crop.py index b4b4dec..20874d7 100644 --- a/mdata/crop.py +++ b/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 -- libgit2 0.21.2