Ignore:
Timestamp:
Jul 14, 2016, 11:54:26 AM (3 years ago)
Author:
dferreira
Message:

Changes made to the tests of illustration2Vec

File:
1 edited

Legend:

Unmodified
Added
Removed
  • internals/2016/aptoideimagesdetector/trunk/Source Code/Illustration2Vector/illustration2vec-master/test.py

    r16289 r16291  
    2323
    2424                img = Image.open(image_file)
    25                 list = illust2vec.estimate_plausible_tags([img], threshold=0.5)
     25                list = illust2vec.estimate_specific_tags([img], ["explicit", "safe"])
    2626                print "Time processing image: "+str(datetime.now()-now)
    2727                f.write('\n')
     
    2929                f.write(image_file)
    3030                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')
     31                if(list[0]['explicit']>list[0]['safe']):
     32                        print "explicit"
     33                        f.write("explicit")
     34                else:
     35                        print "safe"
     36                        f.write("safe")
    3837
    3938for image_dir in ('aptoide_images/nude', 'aptoide_images/not_nude'):
     
    4443
    4544                img = Image.open(image_file)
    46                 list = illust2vec.estimate_plausible_tags([img], threshold=0.5)
     45                list = illust2vec.estimate_specific_tags([img], ["explicit", "safe"])
    4746                print "Time processing image: "+str(datetime.now()-now)
    4847                f.write('\n')
     
    5049                f.write(image_file)
    5150                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')
     51                if(list[0]['explicit']>list[0]['safe']):
     52                        print "explicit"
     53                        f.write("explicit")
     54                else:
     55                        print "safe"
     56                        f.write("safe")
    5957
    6058print "Time processing all images: "+str(datetime.now()-global_time)
Note: See TracChangeset for help on using the changeset viewer.