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

Chunk
1 parent 4e8ab1ef
Showing 2 changed files with 3 additions and 1 deletions Show diff stats
mdata/ILSVRC.py
mdata/ILSVRC_S.py
@@ -294,7 +294,7 @@ class DataILSVRC(DataDumperBase):
                     # img = cv2.imread(image, cv2.CV_LOAD_IMAGE_UNCHANGED)
                     # h, w = img.shape[:2]
                     # if w < 300 or h < 300:
-                    #         continue
+                    # continue
                     #     left, upper = random.randint(0, w - 300), random.randint(0, h - 300)
                     #     img_crop = img[upper:upper + 300, left:left + 300]
                     #     cv2.imwrite(os.path.join(base_dir, category + '_crop_cv', name), img_crop)
@@ -457,6 +457,7 @@ class DataILSVRC(DataDumperBase):
                             dict_dataset[hash] = (tag, im.getCoefMatrix(channel='Y'))
  
                 for tag, feat in dict_dataset.values():
+                    feat.ravel()[[i*304+j for i in range(0,304,8) for j in range(0,304,8)]] = 0
                     X.append(feat.ravel())
                     Y.append(int(tag))
  
@@ -520,6 +520,7 @@ class DataILSVRC_S(DataDumperBase):
                             dict_dataset[hash] = (tag, im.getCoefMatrix(channel='Y'))
  
                 for tag, feat in dict_dataset.values():
+                    feat.ravel()[[i*304+j for i in range(0,304,8) for j in range(0,304,8)]] = 0
                     X.append(feat.tolist())
                     Y.append(int(tag))
...	...	@@ -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
...	...
...	...	@@ -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
...	...