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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
- import os, shutil
- import conf
- # The path to the directory where the original
- # dataset was uncompressed
- original_dataset_dir = os.path.join(conf.data_dir, 'train')
- # The directory where we will
- # store our smaller dataset
- os.mkdir(conf.base_dir)
- # Directories for our training,
- # validation and test splits
- os.mkdir(conf.train_dir)
- os.mkdir(conf.validation_dir)
- os.mkdir(conf.test_dir)
- # Directory with our training cat pictures
- train_cats_dir = os.path.join(conf.train_dir, 'cats')
- os.mkdir(train_cats_dir)
- # Directory with our training dog pictures
- train_dogs_dir = os.path.join(conf.train_dir, 'dogs')
- os.mkdir(train_dogs_dir)
- # Directory with our validation cat pictures
- validation_cats_dir = os.path.join(conf.validation_dir, 'cats')
- os.mkdir(validation_cats_dir)
- # Directory with our validation dog pictures
- validation_dogs_dir = os.path.join(conf.validation_dir, 'dogs')
- os.mkdir(validation_dogs_dir)
- # Directory with our validation cat pictures
- test_cats_dir = os.path.join(conf.test_dir, 'cats')
- os.mkdir(test_cats_dir)
- # Directory with our validation dog pictures
- test_dogs_dir = os.path.join(conf.test_dir, 'dogs')
- os.mkdir(test_dogs_dir)
- # Copy first 1000 cat images to train_cats_dir
- fnames = ['cat.{}.jpg'.format(i) for i in range(1000)]
- for fname in fnames:
- src = os.path.join(original_dataset_dir, fname)
- dst = os.path.join(train_cats_dir, fname)
- shutil.copyfile(src, dst)
- # Copy next 500 cat images to validation_cats_dir
- fnames = ['cat.{}.jpg'.format(i) for i in range(1000, 1500)]
- for fname in fnames:
- src = os.path.join(original_dataset_dir, fname)
- dst = os.path.join(validation_cats_dir, fname)
- shutil.copyfile(src, dst)
-
- # Copy next 500 cat images to test_cats_dir
- fnames = ['cat.{}.jpg'.format(i) for i in range(1500, 2000)]
- for fname in fnames:
- src = os.path.join(original_dataset_dir, fname)
- dst = os.path.join(test_cats_dir, fname)
- shutil.copyfile(src, dst)
-
- # Copy first 1000 dog images to train_dogs_dir
- fnames = ['dog.{}.jpg'.format(i) for i in range(1000)]
- for fname in fnames:
- src = os.path.join(original_dataset_dir, fname)
- dst = os.path.join(train_dogs_dir, fname)
- shutil.copyfile(src, dst)
-
- # Copy next 500 dog images to validation_dogs_dir
- fnames = ['dog.{}.jpg'.format(i) for i in range(1000, 1500)]
- for fname in fnames:
- src = os.path.join(original_dataset_dir, fname)
- dst = os.path.join(validation_dogs_dir, fname)
- shutil.copyfile(src, dst)
-
- # Copy next 500 dog images to test_dogs_dir
- fnames = ['dog.{}.jpg'.format(i) for i in range(1500, 2000)]
- for fname in fnames:
- src = os.path.join(original_dataset_dir, fname)
- dst = os.path.join(test_dogs_dir, fname)
- shutil.copyfile(src, dst)
|