def build_distribution(est, n_clusters, resize_ratio, resize_method, crop_amount):
print 'BuildingDistribution'
images = imagesHandler.get_all_img_rows()
allDist = []
numimg = 0
for image in images:
if numimg%100==0:
print numimg
imageId = image[0]
imageURL = image[1]
img = cv2.imread(dirm.rootDirectory + imageURL)
#print dirm.rootDirectory + imageURL
try:
kp,desc,gray = interest_point_detectors.calculate_orb_values(img, resize_ratio, resize_method, crop_amount)
except Exception,e:
print "unable to process image "+imageId
print str(e)
dist = np.zeros((n_clusters))
imagePred = est.predict(desc)
for p in imagePred:
dist[p] = dist[p] + 1
dist = dist / len(imagePred)
allDist.append([imageId] + dist.tolist())
numimg = numimg +1
def extract_all_descriptors(hessian_threshold, resize_ratio, resize_method,
crop_amount):
print 'Extracting Discriptors: SURF'
images = imagesHandler.get_all_img_rows()
concatDesc = []
concatIds = []
numimg = 0
numDesc = 0
for image in images:
if numimg % 100 == 0:
print numimg
imageId = image[0]
imageURL = image[1]
img = cv2.imread(dirm.rootDirectory + imageURL)
try:
kp, desc, gray = interest_point_detectors.calculate_surf_values(
img, hessian_threshold, resize_ratio, resize_method,
crop_amount)
numDesc = numDesc + len(desc)
for d in desc:
concatDesc.append(d)
concatIds.append([imageId, len(desc)])
except Exception, e:
print "unable to process image " + imageId
print str(e)
numimg = numimg + 1
def extract_all_descriptors(resize_ratio, resize_method, crop_amount):
print 'Extracting Discriptors: orb'
images = imagesHandler.get_all_img_rows()
concatDesc = []
concatIds = []
numimg = 0
numDesc = 0
for image in images:
if numimg%100==0:
print numimg
imageId = image[0]
imageURL = image[1]
img = cv2.imread(dirm.rootDirectory + imageURL)
try:
kp,desc,gray = interest_point_detectors.calculate_orb_values(img, resize_ratio, resize_method, crop_amount)
numDesc = numDesc + len(desc)
for d in desc:
concatDesc.append(d)
concatIds.append([imageId,len(desc)])
except Exception,e:
print "unable to process image "+imageId
print str(e)
numimg = numimg +1
