-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetupColab.py
17 lines (12 loc) · 1.11 KB
/
setupColab.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from broccole.CocoDataset import CocoDataset
from broccole.CocoDatasetBuilder import CocoDatasetBuilder
from broccole.logUtils import init_logging
init_logging('setupColab.log')
humanDataset = CocoDatasetBuilder('../drive/My Drive/broccole/data/annotations/instances_val2017.json', '../val2017').addClasses([1]).build()
CocoDataset.save(humanDataset, '../drive/My Drive/broccole/data/human')
nonHumanDataset = CocoDatasetBuilder('../drive/My Drive/broccole/data/annotations/instances_val2017.json', '../val2017').selectAll().filterNonClasses([1]).build(shuffle=True)
CocoDataset.save(nonHumanDataset, '../drive/My Drive/broccole/data/nonHuman')
valHumanDataset = CocoDatasetBuilder('../drive/My Drive/broccole/data/annotations/instances_val2017.json', '../val2017').addClasses([1]).build()
CocoDataset.save(valHumanDataset, '../drive/My Drive/broccole/data/valHuman')
valNonHumanDataset = CocoDatasetBuilder('../drive/My Drive/broccole/data/annotations/instances_val2017.json', '../val2017').selectAll().filterNonClasses([1]).build(shuffle=True)
CocoDataset.save(valNonHumanDataset, '../drive/My Drive/broccole/data/valNonHuman')