luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

generate raw curv values for test set

Browse Source
This commit is contained in:
luni
2023-01-28 16:18:29 +00:00
parent a5659196a2
commit 632d8db815
2 changed files with 17 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
curvature_label_parallel.py
+4 -4
View File
@@ -4,16 +4,16 @@ import os
if __name__ == '__main__': if __name__ == '__main__':
folder = '/home/workspace/data/dataset/local-cache/train_skirt' folder = '/home/workspace/data/dataset/test_skirt_ready_2'
out_folder = '/home/workspace/local-data/valid_skirt' out_folder = '/home/workspace/local-data/valid_skirt'
ids = os.listdir(folder + '/mesh_posed') ids = os.listdir(folder + '/mesh_posed')
count = 0 count = 0
step = 5000 step = 5000
# while count < len(ids): while count < len(ids):
while count < 30000: # while count < 30000:
# os.system('nohup python MeshUtils/curvature_labels.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + ' -n ' + 'curv_train_' + str(count) + '_' + str(count+step) + ' > curv_train_' + str(count) + '_' + str(count+step) + '.out' + ' 2>&1 &') # os.system('nohup python MeshUtils/curvature_labels.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + ' -n ' + 'curv_train_' + str(count) + '_' + str(count+step) + ' > curv_train_' + str(count) + '_' + str(count+step) + '.out' + ' 2>&1 &')
os.system('nohup python curvature_labels.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + ' -n ' + 'curv_gen_10_train_skirt_' + str(count) + '_' + str(count+step) + ' > nohup/curv_gen_10_train_skirt_' + str(count) + '_' + str(count+step) + '.out' + ' 2>&1 &') os.system('nohup python curvature_labels.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + ' -n ' + 'curv_gen_10_test_skirt_' + str(count) + '_' + str(count+step) + ' > nohup/curv_gen_10_test_skirt_' + str(count) + '_' + str(count+step) + '.out' + ' 2>&1 &')
# os.system('python Data/convert_to_uv.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + '') # os.system('python Data/convert_to_uv.py -f ' + folder + ' -s ' + str(count) + ' -e ' + str(count + step) + '')
count += step count += step
curvature_labels.py
+13 -13
View File
@@ -251,9 +251,9 @@ def read_raw_curvatures(folder, frames, neighbourhood):
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("-f", "--folder", help = "process the given folder", default=r'/home/workspace/data/dataset/local-cache/train_skirt') parser.add_argument("-f", "--folder", help = "process the given folder", default=r'/home/workspace/data/dataset/test_skirt_ready_2')
parser.add_argument("-s", "--start", help = "start of range", default=0) parser.add_argument("-s", "--start", help = "start of range", default=5000)
parser.add_argument("-e", "--end", help = "end of range", default=30000) parser.add_argument("-e", "--end", help = "end of range", default=10000)
parser.add_argument("-n", "--name", help = "end of range", default="lhunor") parser.add_argument("-n", "--name", help = "end of range", default="lhunor")
args = parser.parse_args() args = parser.parse_args()
@@ -275,20 +275,20 @@ if __name__ == '__main__':
print('SKIRT' if is_skirt else 'NOT SKIRT') print('SKIRT' if is_skirt else 'NOT SKIRT')
# frames = sorted(os.listdir(os.path.join(folder, 'mesh_posed')))[int(args.start):int(args.end)] frames = sorted(os.listdir(os.path.join(folder, 'mesh_posed')))[int(args.start):int(args.end)]
frames = [] # frames = []
with open(os.path.join(folder, frames_file), 'r') as f: # with open(os.path.join(folder, frames_file), 'r') as f:
lines = f.readlines() # lines = f.readlines()
lines = [line.strip() for line in lines] # lines = [line.strip() for line in lines]
frames = lines[int(args.start):int(args.end)] # frames = lines[int(args.start):int(args.end)]
# generate_curvatures(folder, neighbourhood, frames) generate_curvatures(folder, neighbourhood, frames)
curvatures = read_raw_curvatures(folder, frames, neighbourhood) # curvatures = read_raw_curvatures(folder, frames, neighbourhood)
# bins = get_bins(curvatures, nr_of_labels) # bins = get_bins(curvatures, nr_of_labels)
# np.save('train_bins_skirt.npy', bins) # np.save('train_bins_skirt.npy', bins)
bins = np.load('custom_data/train_bins_skirt.npy') # bins = np.load('custom_data/train_bins_skirt.npy')
generate_curvature_labels(folder, bins, curvatures, frames, neighbourhood) # generate_curvature_labels(folder, bins, curvatures, frames, neighbourhood)
# visualize_curvature_labels(folder) # visualize_curvature_labels(folder)
pass pass