Skip to content

Commit e95e492

Browse files
evaluation
1 parent ec11ce8 commit e95e492

21 files changed

+1352
-151
lines changed

PlaneSetGeneration/train_planenet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
MOVING_AVERAGE_DECAY = 0.99
3333

3434
deepSupervisionLayers=['res4b22_relu']
35-
deepSupervisionLayers=[]
35+
#deepSupervisionLayers=[]
3636

3737
def build_graph(img_inp_train, img_inp_val, plane_gt_train, plane_gt_val, validating_inp, is_training=True, numOutputPlanes=20, gpu_id = 0, useCRF= 0, suffix='forward'):
3838
if suffix == '12_22':
@@ -731,7 +731,7 @@ def test(gpu_id, dumpdir, logdir, testdir, keyname, restore, numOutputPlanes=20,
731731
ranges = np.array([urange / imageWidth * 640 / focalLength, np.ones(urange.shape), -vrange / imageHeight * 480 / focalLength]).transpose([1, 2, 0])
732732

733733

734-
for index in xrange(100):
734+
for index in xrange(10):
735735
print(('image', index))
736736
t0=time.time()
737737
#im, planes, depth, normal, boundary, s_8, p_8, b_8, s_16, p_16, b_16, s_32, p_32, b_32 = sess.run([img_inp, plane_gt, depth_gt, normal_gt, boundary_gt, s_8_gt, p_8_gt, b_8_gt, s_16_gt, p_16_gt, b_16_gt, s_32_gt, p_32_gt, b_32_gt])

code/RecordWriterRGBD.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,10 +100,10 @@ def writeRecordFile(tfrecords_filename, imagePaths):
100100
if __name__=='__main__':
101101
imagePaths = glob.glob('/home/chenliu/Projects/Data/NYU_RGBD/raw/train/color_*.png')
102102
imagePaths = [{'image': imagePath, 'depth': imagePath.replace('color', 'depth')} for imagePath in imagePaths]
103-
#print(len(imagePaths))
103+
print(len(imagePaths))
104104
#exit(1)
105105
random.shuffle(imagePaths)
106-
writeRecordFile('../planes_nyu_rgbd_train_temp.tfrecords', imagePaths)
106+
writeRecordFile('../planes_nyu_rgbd_train.tfrecords', imagePaths)
107107
#reader.readRecordFile()
108108

109109

code/evaluate.py

Lines changed: 395 additions & 0 deletions
Large diffs are not rendered by default.

code/kaffe/__init__.pyc

0 Bytes
Binary file not shown.

code/kaffe/caffe/__init__.pyc

0 Bytes
Binary file not shown.

code/kaffe/caffe/resolver.pyc

0 Bytes
Binary file not shown.

code/kaffe/errors.pyc

0 Bytes
Binary file not shown.

code/kaffe/graph.pyc

0 Bytes
Binary file not shown.

code/kaffe/layers.pyc

0 Bytes
Binary file not shown.

code/kaffe/shapes.pyc

0 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)