Niv Sardi commited on
Commit
c9f2bb9
1 Parent(s): 5a84a2a

python: fix chargingbar names

Browse files
Files changed (2) hide show
  1. python/augment.py +1 -1
  2. python/get_entities.py +1 -1
python/augment.py CHANGED
@@ -117,7 +117,7 @@ def process(args):
117
 
118
  batches.append(UnnormalizedBatch(images=a,heatmaps=h))
119
 
120
- bar = ChargingBar('Processing', max=len(batches))
121
  # We use a single, very fast augmenter here to show that batches
122
  # are only loaded once there is space again in the buffer.
123
  pipeline = pipelines.HUGE
 
117
 
118
  batches.append(UnnormalizedBatch(images=a,heatmaps=h))
119
 
120
+ bar = ChargingBar('augment', max=(len(batches)**2)/3*len(background_images))
121
  # We use a single, very fast augmenter here to show that batches
122
  # are only loaded once there is space again in the buffer.
123
  pipeline = pipelines.HUGE
python/get_entities.py CHANGED
@@ -22,7 +22,7 @@ with open(f'{defaults.MAIN_CSV_PATH}.tmp', 'w', newline='') as csvfile:
22
  writer = csv.writer(csvfile)
23
  writer.writerow(Entity.row_names())
24
 
25
- bar = ChargingBar('Processing', max=len(options))
26
  for o in options[1:]:
27
  def get_bco():
28
  (name, bco)= (o.text, o.attrs['value'])
 
22
  writer = csv.writer(csvfile)
23
  writer.writerow(Entity.row_names())
24
 
25
+ bar = ChargingBar('get entities', max=len(options))
26
  for o in options[1:]:
27
  def get_bco():
28
  (name, bco)= (o.text, o.attrs['value'])