Update app.py

app.py

@@ -9,6 +9,8 @@ from typing import NamedTuple, List, Callable, List, Tuple, Optional
 from torch import nn
 import torch.nn.functional as F
 
+
+
 class LinData(NamedTuple):
     in_dim : int # input dimension
     hidden_layers : List[int] # hidden layers including the output layer
@@ -31,6 +33,25 @@ class CNNData(NamedTuple):
 class NetData(NamedTuple):
     cnn3d : CNNData
     lin : LinData
+
+def conv3D_output_size(args, img_size):
+
+    if not isinstance(args, CNNData):
+            raise TypeError("input must be a ParserClass")
+            
+    (cin, h , w) = img_size
+    # compute output shape of conv3D
+    for idx , chan in enumerate(args.kernel_size):
+        padding = args.paddings[idx]
+        stride = args.strides[idx]
+        (cin, h , w) = (np.floor((cin + 2 * padding[0] - chan[0] ) / stride[0] + 1).astype(int),
+                    np.floor((h + 2 * padding[1] - chan[1] ) / stride[1] + 1).astype(int),
+                    np.floor((w + 2 * padding[2] - chan[2] ) / stride[2] + 1).astype(int))
+        
+        
+    final_dim = int(args.n_f[-1] * cin * h * w)
+    
+    return final_dim
     
 class CNN3D_Mike(nn.Module):
     def __init__(self, t_dim=30, img_x=256 , img_y=342, drop_p=0, fc_hidden1=256, fc_hidden2=256):