@@ -176,12 +176,12 @@ def vggface2_root():
176
176
"""
177
177
Generates a dataset with the following folder structure and returns the path root:
178
178
<root>
179
- └── vggface2
180
- ├── bb_landmark.tar.gz ('bb_landmark' when uncompressed)
181
- ├── vggface2_train.tar.gz ('train' when uncompressed)
182
- ├── vggface2_test.tar.gz ('test' when uncompressed)
183
- ├── train_list.txt
184
- └── test_list.txt
179
+ └── vggface2
180
+ ├── bb_landmark.tar.gz ('bb_landmark' when uncompressed)
181
+ ├── vggface2_train.tar.gz ('train' when uncompressed)
182
+ ├── vggface2_test.tar.gz ('test' when uncompressed)
183
+ ├── train_list.txt
184
+ └── test_list.txt
185
185
186
186
The dataset consist of 1 image in the train set and 1 image in the test set.
187
187
"""
@@ -232,8 +232,8 @@ def _make_test_archive(root):
232
232
_make_tar (test_archive , top_level_dir , arcname = 'test' , compress = True )
233
233
234
234
def _make_bb_landmark_archive (root ):
235
- train_bb_contents = 'NAME_ID,X,Y,W,H\n "n000001/0001_01",161,140,224,324'
236
- test_bb_contents = 'NAME_ID,X,Y,W,H\n "n000001/0001_01",161,140,224,324'
235
+ train_bbox_contents = 'NAME_ID,X,Y,W,H\n "n000001/0001_01",161,140,224,324'
236
+ test_bbox_contents = 'NAME_ID,X,Y,W,H\n "n000001/0001_01",161,140,224,324'
237
237
train_landmark_contents = ('NAME_ID,P1X,P1Y,P2X,P2Y,P3X,P3Y,P4X,P4Y,P5X,P5Y\n '
238
238
'"n000001/0001_01",75.81253,110.2077,103.1778,104.6074,'
239
239
'90.06353,133.3624,85.39182,149.4176,114.9009,144.9259' )
@@ -248,12 +248,12 @@ def _make_bb_landmark_archive(root):
248
248
# bbox training file
249
249
bbox_file = os .path .join (extracted_dir , "loose_bb_train.csv" )
250
250
with open (bbox_file , "w" ) as csv_file :
251
- csv_file .write (train_bb_contents )
251
+ csv_file .write (train_bbox_contents )
252
252
253
253
# bbox testing file
254
254
bbox_file = os .path .join (extracted_dir , "loose_bb_test.csv" )
255
255
with open (bbox_file , "w" ) as csv_file :
256
- csv_file .write (test_bb_contents )
256
+ csv_file .write (test_bbox_contents )
257
257
258
258
# landmark training file
259
259
landmark_file = os .path .join (extracted_dir , "loose_landmark_train.csv" )
0 commit comments