perler commited on
Commit
cd8be0a
1 Parent(s): bc70951

use abs paths for rec

Browse files
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -12,10 +12,10 @@ import spaces
12
 
13
  @spaces.GPU(duration=60 * 3)
14
  def run_on_gpu(input_point_cloud: gr.utils.NamedString,
15
- gen_resolution_global,
16
- padding_factor,
17
- gen_subsample_manifold_iter,
18
- gen_refine_iter):
19
  print('Started inference at {}'.format(datetime.datetime.now()))
20
  print('Inputs:', input_point_cloud, gen_resolution_global, padding_factor,
21
  gen_subsample_manifold_iter, gen_refine_iter)
@@ -28,8 +28,11 @@ def run_on_gpu(input_point_cloud: gr.utils.NamedString,
28
  sys.path.append(os.path.abspath('ppsurf'))
29
  from ppsurf.pps import cli_main
30
 
31
- in_file = 'data/{}'.format(input_point_cloud.name)
32
- out_file = 'results/rec/{}'.format(input_point_cloud.name)
 
 
 
33
 
34
  args = [
35
  'pps.py', 'rec',
 
12
 
13
  @spaces.GPU(duration=60 * 3)
14
  def run_on_gpu(input_point_cloud: gr.utils.NamedString,
15
+ gen_resolution_global: int,
16
+ padding_factor: float,
17
+ gen_subsample_manifold_iter: int,
18
+ gen_refine_iter: int):
19
  print('Started inference at {}'.format(datetime.datetime.now()))
20
  print('Inputs:', input_point_cloud, gen_resolution_global, padding_factor,
21
  gen_subsample_manifold_iter, gen_refine_iter)
 
28
  sys.path.append(os.path.abspath('ppsurf'))
29
  from ppsurf.pps import cli_main
30
 
31
+ in_file = '{}'.format(input_point_cloud.name)
32
+
33
+ # append 'rec' to the input file name
34
+ splitext_result = os.path.splitext(in_file)
35
+ out_file = splitext_result[0] + '_rec' + splitext_result[1]
36
 
37
  args = [
38
  'pps.py', 'rec',