umgefahren commited on
Commit
d055ece
·
1 Parent(s): c0bc340

Revert "Progress bar"

Browse files

This reverts commit a0cf668e52d007a5b5f734a51bd234be1887c833.

Files changed (1) hide show
  1. main.py +1 -8
main.py CHANGED
@@ -53,24 +53,19 @@ generate_tree_images stores the cutout tree images in a separate folder
53
  """
54
 
55
 
56
- def greet(image_path: str, progress=gr.Progress()):
57
  current_directory = os.getcwd()
58
 
59
- progress(0, desc="Starting...")
60
-
61
  output_directory = os.path.join(current_directory, "outputs")
62
  if not os.path.exists(output_directory):
63
  os.makedirs(output_directory)
64
 
65
- progress(0.1, desc="Running detectree2...")
66
  run_detectree2(image_path, store_path=output_directory)
67
 
68
- progress(0.2, desc="Postprocessing...")
69
  processed_output_df = postprocess(output_directory + '/detectree2_delin.geojson', output_directory + '/processed_delin')
70
 
71
  processed_geojson = output_directory + '/processed_delin.geojson'
72
 
73
- progress(0.3, desc="Generating tree images...")
74
  generate_tree_images(processed_geojson, image_path)
75
 
76
  output_folder = './tree_images'
@@ -90,8 +85,6 @@ def greet(image_path: str, progress=gr.Progress()):
90
  processed_output_df['species'] = all_top_3_list
91
 
92
  final_output_path = 'result'
93
-
94
- progress(0.4, desc="Exporting GeoJSON...")
95
  export_geojson(processed_output_df, final_output_path)
96
 
97
  with rasterio.open(image_path) as src:
 
53
  """
54
 
55
 
56
+ def greet(image_path: str):
57
  current_directory = os.getcwd()
58
 
 
 
59
  output_directory = os.path.join(current_directory, "outputs")
60
  if not os.path.exists(output_directory):
61
  os.makedirs(output_directory)
62
 
 
63
  run_detectree2(image_path, store_path=output_directory)
64
 
 
65
  processed_output_df = postprocess(output_directory + '/detectree2_delin.geojson', output_directory + '/processed_delin')
66
 
67
  processed_geojson = output_directory + '/processed_delin.geojson'
68
 
 
69
  generate_tree_images(processed_geojson, image_path)
70
 
71
  output_folder = './tree_images'
 
85
  processed_output_df['species'] = all_top_3_list
86
 
87
  final_output_path = 'result'
 
 
88
  export_geojson(processed_output_df, final_output_path)
89
 
90
  with rasterio.open(image_path) as src: