Skip to content

Commit ccc4423

Browse files
Update data_converter.py
1 parent 6456d71 commit ccc4423

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

pytorch/tf/data_converter.py

+12-12
Original file line numberDiff line numberDiff line change
@@ -71,18 +71,18 @@ def writeRecordFile(split, dataset):
7171
pass
7272
annotationPath = '/'.join(tokens[:-2]) + '/annotation_new/'
7373

74-
info = global_gt['info'][batchIndex]
75-
info[1] = info[5]
76-
info[3] = info[6]
77-
info[4] = info[16]
78-
info[5] = info[17]
79-
info[6] = info[18]
80-
info[7] = info[8] = info[9] = 0
81-
info = info[:10]
82-
if not os.path.exists(annotationPath + '/info.npy'):
83-
np.save(annotationPath + '/info.npy', info)
84-
pass
85-
continue
74+
# info = global_gt['info'][batchIndex]
75+
# info[1] = info[5]
76+
# info[3] = info[6]
77+
# info[4] = info[16]
78+
# info[5] = info[17]
79+
# info[6] = info[18]
80+
# info[7] = info[8] = info[9] = 0
81+
# info = info[:10]
82+
# if not os.path.exists(annotationPath + '/info.npy'):
83+
# np.save(annotationPath + '/info.npy', info)
84+
# pass
85+
# continue
8686

8787
if os.path.exists(annotationPath + tokens[-1].replace('color.jpg', 'planes.npy')):
8888
continue

0 commit comments

Comments
 (0)