model.py 270 B

123456789
  1. from os import walk, path
  2. def models(models_home='.', *args, **kwargs):
  3. for root, _, files in walk(models_home):
  4. for file in files:
  5. base, ext = path.splitext(file)
  6. if ext == '.bin':
  7. yield base, path.join(root, file)