1 2
__author__ = 'chunk'
3
from ..common import *
4
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
from ..mdata import MSR, CV, ILSVRC def test_MSR(): dmsr = MSR.DataMSR() # msrd.format() # msrd.build_list() # dmsr.store_image() # dmsr.store_tag() # dmsr.extract_feat(feattype='ibd') dmsr.store_feat(feattype='ibd') def test_CV(): dcv = CV.DataCV() # dcv.format() # dcv.build_list() # dcv.get_feat() # dcv.extract_feat() print dcv.get_feat("/home/hadoop/data/HeadShoulder/dst/Train/Img/132/7c5fe33bd194fc1ae7b0023956ebd.jpg", 'ibd') X, Y = dcv.load_data() print len(X), len(Y)
31
def test_ILSVRC():
32
timer = Timer()
33
# dil = ILSVRC.DataILSVRC(base_dir='/data/hadoop/ImageNet/ILSVRC/ILSVRC2013_DET_val', category='Train')
34
dil = ILSVRC.DataILSVRC(base_dir='/data/hadoop/ImageNet/ILSVRC/ILSVRC2013_DET_val', category='Train_1')
35
# dil = ILSVRC.DataILSVRC(base_dir='/media/chunk/Elements/D/data/ImageNet/img/ILSVRC2013_DET_val', category='Train_1')
36
37
dil.format()
38
dil.embed(rate=0.1)
39
dil.extract_feat(feattype='ibd')
40
# dil.extract_feat(feattype='hog')
41
42
timer.mark()
43
dil.store_img()
44 45 46
timer.report() timer.mark()
47
dil.store_tag()
48 49
timer.report()
50
51
dil.store_info()
52
53
54
55
dil.store_feat()
56 57
58 59 60
if __name__ == '__main__': # test_MSR()
61
# test_CV()
62 63 64
test_ILSVRC() print 'helllo'
