-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathprepare_datasets.py
48 lines (34 loc) · 1.75 KB
/
prepare_datasets.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
from utils import buildObjImagesFolder, copyAllImages, buildObjDataFile, buildObjNamesFile, log
log('Construindo arquivo obj.data CCPD_BASE...')
buildObjDataFile(path=['data_ccpd_base', 'obj.data'])
log('Construindo arquivo obj.data CCPD_CHALLENGE...')
buildObjDataFile(path=['data_ccpd_weather', 'obj.data'])
log('Construindo arquivo obj.data CCPD_WEATHER...')
buildObjDataFile(path=['data_ccpd_challenge', 'obj.data'])
log('Construindo arquivo obj.names para CCPD_BASE...')
buildObjNamesFile(path='data_ccpd_base')
log('Construindo arquivo obj.names para CCPD_CHALLENGE...')
buildObjNamesFile(path='data_ccpd_challenge')
log('Construindo arquivo obj.names para CCPD_WEATHER...')
buildObjNamesFile(path='data_ccpd_weather')
log('Construindo arquivo train.txt para CCPD_BASE')
buildObjImagesFolder(
type='train', pathToCopy='datasets/ccpd_base', pathToWrite='data_ccpd_base')
log('Construindo arquivo valid.txt para CCPD_BASE')
buildObjImagesFolder(type='valid', pathToCopy='datasets/ccpd_base',
pathToWrite='data_ccpd_base')
log('Construindo arquivo valid.txt para CCPD_CHALLENGE')
buildObjImagesFolder(type='valid', pathToCopy='datasets/ccpd_challenge',
pathToWrite='data_ccpd_challenge')
log('Construindo arquivo valid.txt para CCPD_WEATHER')
buildObjImagesFolder(
type='valid', pathToCopy='datasets/ccpd_weather', pathToWrite='data_ccpd_weather')
log('Copiando imagens para CCPD_BASE')
copyAllImages(path='data_ccpd_base',
datasetFolder='ccpd_base')
log('Copiando imagens para CCPD_CHALLENGE')
copyAllImages(path='data_ccpd_challenge',
datasetFolder='ccpd_challenge')
log('Copiando imagens para CCPD_WEATHER')
copyAllImages(path='data_ccpd_weather',
datasetFolder='ccpd_weather')