Chunk / ImageR

Download as
Browse Code ยป

Commit 25c0c9c9640afd5d75af3d575c2cd81d2f5dc8ee

Authored by Chunk
1 parent 4e8ab1ef
Exists in master and in 1 other branch refactor

feat.ravel()[[i*304+j for i in range(0,304,8) for j in range(0,304,8)]] = 0

Showing 2 changed files with 3 additions and 1 deletions   Show diff stats
mdata/ILSVRC.py
  Diff comments   View file @25c0c9c
... ... @@ -294,7 +294,7 @@ class DataILSVRC(DataDumperBase):
294 294 # img = cv2.imread(image, cv2.CV_LOAD_IMAGE_UNCHANGED)
295 295 # h, w = img.shape[:2]
296 296 # if w < 300 or h < 300:
297   - # continue
  297 + # continue
298 298 # left, upper = random.randint(0, w - 300), random.randint(0, h - 300)
299 299 # img_crop = img[upper:upper + 300, left:left + 300]
300 300 # cv2.imwrite(os.path.join(base_dir, category + '_crop_cv', name), img_crop)
... ... @@ -457,6 +457,7 @@ class DataILSVRC(DataDumperBase):
457 457 dict_dataset[hash] = (tag, im.getCoefMatrix(channel='Y'))
458 458  
459 459 for tag, feat in dict_dataset.values():
  460 + feat.ravel()[[i*304+j for i in range(0,304,8) for j in range(0,304,8)]] = 0
460 461 X.append(feat.ravel())
461 462 Y.append(int(tag))
462 463  
... ...
mdata/ILSVRC_S.py
  Diff comments   View file @25c0c9c
... ... @@ -520,6 +520,7 @@ class DataILSVRC_S(DataDumperBase):
520 520 dict_dataset[hash] = (tag, im.getCoefMatrix(channel='Y'))
521 521  
522 522 for tag, feat in dict_dataset.values():
  523 + feat.ravel()[[i*304+j for i in range(0,304,8) for j in range(0,304,8)]] = 0
523 524 X.append(feat.tolist())
524 525 Y.append(int(tag))
525 526  
... ...