Update depth_app.py

depth_app.py

@@ -7,8 +7,11 @@ import io
 from torchvision import transforms
 import matplotlib as mpl
 import matplotlib.cm as cm
-import networks
 from layers import disp_to_depth_no_scaling
+from resnet_encoder import ResnetEncoder
+from depth_decoder import DepthDecoder
+# from pose_decoder import PoseDecoder
+# from pose_cnn import PoseCNN
 
 # Function to load the model
 def load_model(device, model_path):
@@ -16,7 +19,7 @@ def load_model(device, model_path):
     encoder_path = os.path.join(model_path, "encoder.pth")
     depth_decoder_path = os.path.join(model_path, "depth.pth")
 
-    encoder = networks.ResnetEncoder(18, False)
+    encoder = ResnetEncoder(18, False)
     loaded_dict_enc = torch.load(encoder_path, map_location=device)
     feed_height = loaded_dict_enc['height']
     feed_width = loaded_dict_enc['width']
@@ -25,7 +28,7 @@ def load_model(device, model_path):
     encoder.to(device)
     encoder.eval()
 
-    depth_decoder = networks.DepthDecoder(num_ch_enc=encoder.num_ch_enc, scales=range(4))
+    depth_decoder = DepthDecoder(num_ch_enc=encoder.num_ch_enc, scales=range(4))
     loaded_dict = torch.load(depth_decoder_path, map_location=device)
     depth_decoder.load_state_dict(loaded_dict, strict=False)
     depth_decoder.to(device)