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='/media/chunk/Elements/D/data/ImageNet/img/ILSVRC2013_DET_val', category='Test')
35
dil.format()
36
37
# dil.extract_feat(feattype='ibd')
38
# dil.extract_feat(feattype='hog')
39
40
# timer.mark()
41
# dil.store_img()
42
# timer.report()
43
#
44 45 46
# timer.mark() # dil.store_info() # timer.report()
47
48 49
if __name__ == '__main__':
50
# test_MSR()
51
# test_CV()
52
test_ILSVRC()
53
54
print 'helllo'