make_det_boxes.py 533 B

1234567891011121314151617181920212223
  1. '''
  2. name 98, 200, 380, 250
  3. birth 98, 297, 420, 330
  4. sex 570, 305, 600, 338
  5. addr 98, 375, 420, 400
  6. id 445, 370, 635, 405
  7. '''
  8. import argparse
  9. import os
  10. def write_det_boxes(img_dir):
  11. img_list = os.listdir(img_dir)
  12. for img_path in img_list:
  13. img_path = str(img_dir).split('\\')[-1] + '/' + img_path
  14. print(img_path)
  15. if __name__ == "__main__":
  16. parser = argparse.ArgumentParser()
  17. parser.add_argument('--img_dir', type=str, default='./imgs')
  18. args = parser.parse_args()
  19. write_det_boxes(args.img_dir)