Advertisement
Guest User

Untitled

a guest
Jul 17th, 2019
80
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.30 KB | None | 0 0
  1. root = '101_ObjectCategories'
  2. exclude = ['BACKGROUND_Google', 'Motorbikes', 'airplanes', 'Faces_easy', 'Faces']
  3. train_split, val_split = 0.7, 0.15
  4.  
  5. categories = [x[0] for x in os.walk(root) if x[0]][1:]
  6. categories = [c for c in categories if c not in [os.path.join(root, e) for e in exclude]]
  7.  
  8. print(categories)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement