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

generate test curv

Browse Source
This commit is contained in:
luni
2023-01-28 20:25:31 +00:00
parent b121ee25e1
commit 28cdd79a6b
2 changed files with 13 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
aaa_helpers.txt
+2 -1
View File
@@ -1,4 +1,5 @@
nohup python curvature_labels.py > nohup/curv_labels_train_skirt_15k_25k.out 2>&1 & nohup python curvature_labels.py > nohup/curv_labels_test_skirt_0k_30k.out 2>&1 &
nohup python curvature_labels.py > nohup/curv_labels_valid_skirt_label_0k_30k.out 2>&1 &
one final convolutional layer for smoothing, on the final coordinates one final convolutional layer for smoothing, on the final coordinates
curvature_labels.py
+10 -10
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=15000) parser.add_argument("-s", "--start", help = "start of range", default=0)
parser.add_argument("-e", "--end", help = "end of range", default=25000) parser.add_argument("-e", "--end", help = "end of range", default=30000)
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()
@@ -276,14 +276,14 @@ if __name__ == '__main__':
print('SKIRT' if is_skirt else 'NOT SKIRT') print('SKIRT' if is_skirt else 'NOT SKIRT')
print(f'params: -f {args.folder} -s {args.start} -e {args.end} -n {args.name} -skirt {is_skirt} -neigh {neighbourhood}') print(f'params: -f {args.folder} -s {args.start} -e {args.end} -n {args.name} -skirt {is_skirt} -neigh {neighbourhood}')
# 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 = [frame.split('.')[0] for frame in frames] frames = [frame.split('.')[0] for frame in frames]
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)