source: internals/2016/aptoideimagesdetector/trunk/Source Code/nude.py-master/nude.py-master/tests/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.

File size: 1.2 KB
Line 
1import os
2from datetime import datetime, time
3import glob
4from nude import Nude
5
6ROOT = os.path.dirname(os.path.abspath(__file__))
7
8fp=0
9p=0
10n=0
11fn=0
12
13for image_dir in ('images/not_nude', 'images/nude'):
14        for image_file in glob.glob(os.path.join(image_dir,"*.png")):
15                print image_file
16                now = datetime.now()
17                nu = Nude(os.path.join(ROOT, image_file))
18                nu.parse()
19                if(image_dir=='images/not_nude'):
20                        if nu.result==False:
21                                n+=1
22                        else:
23                                p+=1
24                                fp+=1
25                if(image_dir=='images/nude'):
26                        if nu.result==True:
27                                p+=1
28                        else:
29                                n+=1
30                                fn+=1
31                print(nu.result, nu.inspect())
32                print "Time processing image: "+str(datetime.now()-now)
33
34for image_dir in ('images/not_nude', 'images/nude'):
35        for image_file in glob.glob(os.path.join(image_dir,"*.jpg")):
36                print image_file
37                now = datetime.now()
38                nu = Nude(os.path.join(ROOT, image_file))
39                nu.parse()
40                if(image_dir=='images/not_nude'):
41                        if nu.result==False:
42                                n+=1
43                        else:
44                                p+=1
45                                fp+=1
46                if(image_dir=='images/nude'):
47                        if nu.result==True:
48                                p+=1
49                        else:
50                                n+=1
51                                fn+=1
52                print(nu.result, nu.inspect())
53                print "Time processing image: "+str(datetime.now()-now)
54
55print "n: "+str(n)
56print "p: "+str(p)
57print "fp: "+str(fp)
58print "fn: "+str(fn)
Note: See TracBrowser for help on using the repository browser.