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

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

Changes made to the tests of illustration2Vec

  • Property svn:executable set to *
File size: 1.5 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_specific_tags([img], ["explicit", "safe"])
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                if(list[0]['explicit']>list[0]['safe']):
32                        print "explicit"
33                        f.write("explicit")
34                else:
35                        print "safe"
36                        f.write("safe")
37
38for image_dir in ('aptoide_images/nude', 'aptoide_images/not_nude'):
39        for image_file in glob.glob(os.path.join(image_dir,"*.jpg")):
40               
41                print "Processing image..."+str(image_file)
42                now = datetime.now()
43
44                img = Image.open(image_file)
45                list = illust2vec.estimate_specific_tags([img], ["explicit", "safe"])
46                print "Time processing image: "+str(datetime.now()-now)
47                f.write('\n')
48                f.write('\n')
49                f.write(image_file)
50                f.write('\n')
51                if(list[0]['explicit']>list[0]['safe']):
52                        print "explicit"
53                        f.write("explicit")
54                else:
55                        print "safe"
56                        f.write("safe")
57
58print "Time processing all images: "+str(datetime.now()-global_time)
59
60
61f.close()
Note: See TracBrowser for help on using the repository browser.