New branch for processing RGBD images - additional changes - using bigger dataset and added batchnorm to increse learning speed and prevent overfitting

Author: savnani5

__pycache__/dataloader.cpython-38.pyc

@@ -119,20 +119,6 @@ def __call__(self, image_tensor):
     #         # cv2.waitKey(0)
     #         break
 
-    #____________________________Checking Rescale class__________________________________
-    # scale = Rescale((400, 400))
-    # composed = transforms.Compose([scale])
-    # fig = plt.figure()
-    # sample = sd[3]
-    # for i, tsfrm in enumerate([scale, composed]):
-    #     transformed_sample = tsfrm(sample)
-
-    #     ax = plt.subplot(1, 3, i + 1)
-    #     plt.tight_layout()
-    #     ax.set_title(type(tsfrm).__name__)
-    #     plt.imshow(transformed_sample['image'])
-
-    # plt.show()
 
     #___________________________Checking the iteration on dataset_________________________
     # for i in range(len(dataset)):