evaluation

Author: art-programmer

PlaneSetGeneration/train_planenet.py

@@ -32,7 +32,7 @@
 MOVING_AVERAGE_DECAY = 0.99
 
 deepSupervisionLayers=['res4b22_relu']
-deepSupervisionLayers=[]
+#deepSupervisionLayers=[]
 
 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'):
     if suffix == '12_22':
@@ -731,7 +731,7 @@ def test(gpu_id, dumpdir, logdir, testdir, keyname, restore, numOutputPlanes=20,
             ranges = np.array([urange / imageWidth * 640 / focalLength, np.ones(urange.shape), -vrange / imageHeight * 480 / focalLength]).transpose([1, 2, 0])
 
 
-            for index in xrange(100):
+            for index in xrange(10):
                 print(('image', index))
                 t0=time.time()
                 #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

@@ -100,10 +100,10 @@ def writeRecordFile(tfrecords_filename, imagePaths):
 if __name__=='__main__':
     imagePaths = glob.glob('/home/chenliu/Projects/Data/NYU_RGBD/raw/train/color_*.png')
     imagePaths = [{'image': imagePath, 'depth': imagePath.replace('color', 'depth')} for imagePath in imagePaths]
-    #print(len(imagePaths))
+    print(len(imagePaths))
     #exit(1)
     random.shuffle(imagePaths)
-    writeRecordFile('../planes_nyu_rgbd_train_temp.tfrecords', imagePaths)
+    writeRecordFile('../planes_nyu_rgbd_train.tfrecords', imagePaths)
     #reader.readRecordFile()