vfh_recognizer_fs: vfh_recognizer_fs: process_training

00001 import sys
00002 import os
00003 import pickle
00004 
00005 directory = sys.argv[1]
00006 scale_factor = sys.argv[2]
00007 view_file = directory + '/' + 'last_view.txt'
00008 output = open(view_file, 'wb')
00009 pickle.dump(0, output)
00010 output.close()
00011 
00012 #iterate directory
00013 model_id=0
00014 for f in os.listdir(directory):
00015         if os.path.isfile(os.path.join(directory, f)) and f.endswith(".ply"):
00016                 print f
00017                 print f.replace(".ply","")
00018                 model_id_str = f.replace(".ply","")
00019                 #generate views
00020 
00021                 print "rosrun vfh_recognizer_fs generate_views_vfh.py " + os.path.join(directory, f) + " " + model_id_str + " " + str(0.001) 
00022                 os.system("rosrun vfh_recognizer_fs generate_views_vfh.py " + os.path.join(directory, f) + " " + model_id_str + " " + str(scale_factor) )
00023                 model_id = model_id+1
00024
process_training_data.py
