source: internals/2016/aptoideimagesdetector/trunk/Source Code/Illustration2Vector/illustration2vec-master/test.py @ 16289

Last change on this file since 16289 was 16289, checked in by dferreira, 3 years ago

Initial content. All the tests done to three open-source platrofms.

  • Property svn:executable set to *
File size: 1.6 KB
Line 
1import i2v
2from datetime import datetime, time
3from PIL import Image
4import os
5import glob
6
7
8f = open('test.txt','w')
9
10print "Loading tags..."
11now = datetime.now()
12illust2vec = i2v.make_i2v_with_chainer(
13    "illust2vec_tag_ver200.caffemodel", "tag_list.json")
14print "Time loading tags: "+str(datetime.now()-now)
15
16global_time = datetime.now()
17
18for image_dir in ('aptoide_images/nude', 'aptoide_images/not_nude'):
19        for image_file in glob.glob(os.path.join(image_dir,"*.png")):
20               
21                print "Processing image..."+str(image_file)
22                now = datetime.now()
23
24                img = Image.open(image_file)
25                list = illust2vec.estimate_plausible_tags([img], threshold=0.5)
26                print "Time processing image: "+str(datetime.now()-now)
27                f.write('\n')
28                f.write('\n')
29                f.write(image_file)
30                f.write('\n')
31                for item in list:
32                        for char in item.values():
33                                for one in char:
34                                        for fin in one:
35                                                print fin
36                                                f.write(str(fin))
37                                                f.write('\n')
38
39for image_dir in ('aptoide_images/nude', 'aptoide_images/not_nude'):
40        for image_file in glob.glob(os.path.join(image_dir,"*.jpg")):
41               
42                print "Processing image..."+str(image_file)
43                now = datetime.now()
44
45                img = Image.open(image_file)
46                list = illust2vec.estimate_plausible_tags([img], threshold=0.5)
47                print "Time processing image: "+str(datetime.now()-now)
48                f.write('\n')
49                f.write('\n')
50                f.write(image_file)
51                f.write('\n')
52                for item in list:
53                        for char in item.values():
54                                for one in char:
55                                        for fin in one:
56                                                print fin
57                                                f.write(str(fin))
58                                                f.write('\n')
59
60print "Time processing all images: "+str(datetime.now()-global_time)
61
62
63f.close()
Note: See TracBrowser for help on using the repository browser.