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

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

Improving the automated test

  • Property svn:executable set to *
File size: 1.9 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
18fp = 0
19fn = 0
20p = 0
21n = 0
22
23for image_dir in ('aptoide_images/nude', 'aptoide_images/not_nude'):
24        for image_file in glob.glob(os.path.join(image_dir,"*.png")):
25               
26                print "Processing image..."+str(image_file)
27                now = datetime.now()
28
29                img = Image.open(image_file)
30                list = illust2vec.estimate_specific_tags([img], ["explicit", "safe"])
31                print "Time processing image: "+str(datetime.now()-now)
32                f.write('\n')
33                f.write('\n')
34                f.write(image_file)
35                f.write('\n')
36                if(list[0]['explicit']>list[0]['safe']):
37                        print "explicit"
38                        f.write("explicit")
39                        p+=1
40                        if(image_dir=='aptoide_images/not_nude'):
41                                fp+=1
42                else:
43                        print "safe"
44                        f.write("safe")
45                        n+=1
46                        if(image_dir=='aptoide_images/nude'):
47                                fn+=1
48
49for image_dir in ('aptoide_images/nude', 'aptoide_images/not_nude'):
50        for image_file in glob.glob(os.path.join(image_dir,"*.jpg")):
51               
52                print "Processing image..."+str(image_file)
53                now = datetime.now()
54
55                img = Image.open(image_file)
56                list = illust2vec.estimate_specific_tags([img], ["explicit", "safe"])
57                print "Time processing image: "+str(datetime.now()-now)
58                f.write('\n')
59                f.write('\n')
60                f.write(image_file)
61                f.write('\n')
62                if(list[0]['explicit']>list[0]['safe']):
63                        print "explicit"
64                        f.write("explicit")
65                        p+=1
66                        if(image_dir=='aptoide_images/not_nude'):
67                                fp+=1
68                else:
69                        print "safe"
70                        f.write("safe")
71                        n+=1
72                        if(image_dir=='aptoide_images/nude'):
73                                fn+=1
74
75print "Time processing all"+str(n+p)+"images: "+str(datetime.now()-global_time)
76print str(fp)+" of "+str(p)+" false positives"
77print str(fn)+" of "+str(n)+" false negatives"
78
79f.close()
Note: See TracBrowser for help on using the repository browser.