purl = 'https://www.icoolhunt.com/prey/%d' % pid
#downloads the photo to images/ dir
try:
getImageFromURL(purl, str(pid), 'images/')
except urllib2.HTTPError:
with open('processError.log', 'a') as err:
err.write('%s\t%s\n' % (pid, 'HTTP Error'))
continue
#file to be analyzed
filename = 'images/%d' % pid
#returns rgb colors, count=n
try:
photoRGBColors = colorz(filename, n=3)
except IOError:
with open('processError.log', 'a') as err:
err.write('%s\t%s\n' % (pid, 'image IOError'))
continue
#write to file [pid - R1 - G1 - B1 - R2 - G2 - B2 - R3 - G3 - B3]
fRGB.write("%d\t" % pid + '\t'.join([("%s" % color) for sublist in photoRGBColors for color in sublist])+'\n')
#convert to HSV
photoHSVColors = [convert_color(sRGBColor(*c), HSVColor) for c in photoRGBColors]
#write to file [pid - H1 - S1 - V1 - H2 - S2 - V2 - H3 - S3 - V3]
fHSV.write('%d\t' % pid + '\t'.join(['\t'.join([str(color.hsv_h), str(color.hsv_s), str(color.hsv_v)]) for color in photoHSVColors])+'\n')
fRGB.close()
images = os.listdir('images')
n_clusters = 10
fRGB = open('RGB10colors.txt', 'a')
fHSL = open('HSL10colors.txt', 'a')
for pid in images[:10]:
#write to file [pid - R1 - G1 - B1 - R2 - G2 - B2 - R3 - G3 - B3]
#file to be analyzed
filename = 'images/%s' % pid
try:
photoRGBColors = colorz(filename, n=n_clusters)
except IOError:
with open('processImageError.log', 'a') as err:
err.write('%s\t%s\n' % (pid, 'image IOError'))
continue
#pid, R1,G1,B1, R2,G2,B2, R3,G3,B3 = map(int, prey.strip().split('\t'))
#photoRGBColors = [[R1,G1,B1], [R2,G2,B2], [R3,G3,B3]]
fRGB.write("%s\t" % pid + '\t'.join([("%s" % color) for sublist in photoRGBColors for color in sublist])+'\n')
#convert to HSL
photoHSLColors = [convert_color(sRGBColor(*c, is_upscaled=True), HSLColor) for c in photoRGBColors]
#write to file [pid - H1 - S1 - L1 - H2 - S2 - L2 - H3 - S3 - L3]
