Ignore:
Timestamp:
Jul 25, 2016, 4:05:33 PM (4 years ago)
Author:
dferreira
Message:

Black box v1 done with API

File:
1 edited

Legend:

Unmodified
Added
Removed
  • internals/2016/aptoideimagesdetector/trunk/Source Code/analyse_app.py

    r16391 r16392  
    1212def analyse_app(illust2vec, icons, screenshots, description):
    1313
    14         images = []
     14        try:
     15                f = open("./model_apps_info.pickle", "rb")
     16                classifier = pickle.load(f)
     17                f.close()
     18        except:
     19                print "Serialized objects not found"
     20                exit(0)
    1521
     22        icons = []
     23        screens = []
    1624        for icon in icons:
    17                 images.append(analyse_explicit(illust2vec, icon))
     25                icons.append(analyse_explicit(illust2vec, icon))
    1826
    1927        for scr in screenshots:
    20                 images.append(analyse_explicit(illust2vec, scr))
     28                screens.append(analyse_explicit(illust2vec, scr))
    2129
    22         if len(images)==0:
    23                 return 0
    24         else:
    25                 print images
     30        description = text_cat(description)
     31
     32        def find_features(icon_list, scr_list, description_result):
     33                features = {}
    2634                maximum = 0
    27                 for im in images:
    28                         if im[1]>maximum:
    29                                 maximum = im[1]
     35                flag = 0
     36                safe = 0
     37                if len(icon_list)==0:
     38                        maximum = 0.5
     39                        safe = 0.5
     40                else:
     41                        for icon in icon_list:
     42                                for data in icon:
     43                                        if data[0]=='explicit' and data[1]>maximum:
     44                                                maximum = data[1]
     45                                                flag = 1
     46                                        if data[0]=='safe' and flag==1:
     47                                                safe = data[1]
     48                                                flag = 0
     49                maximum_s = 0
     50                flag = 0
     51                safe_s = 0
    3052
    31                 print maximum
     53                if len(scr_list)==0:
     54                        maximum_s = 0.5
     55                        safe_s = 0.5
     56                else:
     57                        for scr in scr_list:
     58                                for data in scr:
     59                                        if data[0]=='explicit' and data[1]>maximum_s:
     60                                                maximum_s = data[1]
     61                                                flag = 1
     62                                        if data[0]=='safe' and flag==1:
     63                                                safe_s = data[1]
     64                                                flag = 0
    3265
    33                 prob2 = text_cat(description)
    34                 print prob2.prob('exp')
    35                 return 0.7*prob2+0.3*prob1
    36 
     66                features['ic_exp'] = maximum
     67                features['ic_non'] = safe
     68                features['sc_exp'] = maximum_s
     69                features['sc_non'] = safe_s
     70                features['desc_exp'] = description_result.prob('exp')
     71                return features
     72        features = find_features(icons, screens, description)
     73        return classifier.prob_classify(features)
    3774
    3875if __name__=='__main__':
     
    5390        description = ""
    5491
    55         icons.append("./API to download database/images/Explicit search/icons_explicit/7883823.png")
    56         screenshots.append("./API to download database/images/Explicit search/screenshot_explicit/18966461_3_hd.jpg")
     92        icons.append('./API to download database/images/Explicit search/icons_explicit/7677797.png')
     93        screenshots.append('./API to download database/images/Explicit search/screenshot_explicit/4694725_8_hd.png')
    5794        description = "Are you looking for Porn Movies ? If yes, you are at right place. Enjoy porn movies. You can download porn movies also to enjoy later. Tag: Porn, Porno, Sex, Adult, Downloader, sex, xxx, xx"
    58 
    59         print analyse_app(illust2vec, icons, screenshots, description)
     95        dist = analyse_app(illust2vec, icons, screenshots, description)
     96        print dist.prob('exp')
     97       
Note: See TracChangeset for help on using the changeset viewer.