example-apps/PDD/pcb-defect-detection/help_utils/tools.py

   1 # -*- coding: utf-8 -*-
   2 from __future__ import division, print_function, absolute_import
   3 import math
   4 import sys
   5 import os
   6
   7
   8 def view_bar(message, num, total):
   9     rate = num / total
  10     rate_num = int(rate * 40)
  11     rate_nums = math.ceil(rate * 100)
  12     r = '\r%s:[%s%s]%d%%\t%d/%d' % (message, ">" * rate_num, " " * (40 - rate_num), rate_nums, num, total,)
  13     sys.stdout.write(r)
  14     sys.stdout.flush()
  15
  16
  17 def mkdir(path):
  18     if not os.path.exists(path):
  19         os.makedirs(path)