Пример #1
0
    hist_maker = SP_histogram(splits, None, scale_levels, nwords)
    fp.close()
    for filename, im_data in data.iteritems():
        im_data['histogram'] = hist_maker.create_histogram(
            im_data['locs'], im_data['membership'],
            amax(im_data['locs'], 0) + 1)

        print filename
    return data, hist_maker


if __name__ == "__main__":
    np.set_printoptions(precision=3, edgeitems=None)

    opts = ioutil.load_command_line(
        sys.argv[1:],
        keys=['input_data', 'nwords', 'image_dir', 'output_dir'],
        flag_keys=['make_hists', 'do_search', 'both'])
    if opts.has_key('make_hists'):
        data, hist_maker = create_histograms(opts['input_data'],
                                             int(opts['nwords']))
        fp = open(opts['output_data'], 'w')
        pickle.dump(data, fp)
        fp.close()
    elif opts.has_key('do_search'):
        fp = open(opts['input'], 'r')
        data = pickle.load(fp)
        hist_maker = SP_histogram(array((2, 2)), None, 1, int(opts['nwords']))
        result, filenames = search(data, hist_maker)
        filenames = [ope(filename)[0] + ".png" for filename in filenames]
        #        pdb.set_trace()
        if not os.path.exists(opts['output_dir']):
Пример #2
0
        os.remove(out_files[i].name)


def blad(a, b):
    print b
    for v in a:
        c = open(v, 'r').readlines()
        print c


if __name__ == "__main__":
    print sys.argv

    if sys.argv[1] == "--detector":
        import ioutil
        opts = ioutil.load_command_line(
            sys.argv[1:], keys=['csv_filename', 'detector', 'config'])
        command = ['python', opts['detector'], '--config', opts['config']]

        tmp_filenames = []
        scene_list = ioutil.ext_filter(opts['scene_path'], ['.xml'])
        for ovlp in [0.5]:
            for alg_type in ['UNION', 'INTERSECTION']:
                com = copy.copy(command)
                com.extend([
                    '--nms_alg', alg_type, '--overlap',
                    str(ovlp), '--alg_name', alg_type[0:5] + str(ovlp)
                ])
                fp = tempfile.NamedTemporaryFile(mode='w', delete=False)
                tmp_filenames.append(fp.name)
                fp.close()
                merge_func = lambda filenames: ioutil.merge_csv_files(
Пример #3
0
		os.remove(in_files[i].name)
		os.remove(out_files[i].name)	
		
			
def blad(a,b):
	print b
	for v in a:
		c =  open(v,'r').readlines()
		print c
		
if __name__=="__main__":
	print sys.argv
	
	if sys.argv[1] == "--detector":
		import ioutil
		opts = ioutil.load_command_line(sys.argv[1:], keys = ['csv_filename',
				'detector', 'config'])
		command = ['python', opts['detector'], '--config', opts['config']]

		tmp_filenames = []
		scene_list = ioutil.ext_filter(opts['scene_path'],['.xml'])
		for ovlp in [0.5]:
			for alg_type in ['UNION', 'INTERSECTION']:
				com = copy.copy(command)
				com.extend(['--nms_alg',alg_type,'--overlap',str(ovlp),
						'--alg_name', alg_type[0:5]+str(ovlp)])
				fp = tempfile.NamedTemporaryFile(mode='w', delete = False)
				tmp_filenames.append(fp.name)
				fp.close()
				merge_func = lambda filenames: ioutil.merge_csv_files(filenames,
											tmp_filenames[-1])
				multi_process(com,'--scene_list', '--csv_filename', scene_list,