Update preprocessing.py

preprocessing.py

@@ -1,7 +1,7 @@
 import torch
-import numpy as np
 
 def pad_to_22_channels(input_tensor):
     if input_tensor.shape[1] == 3:  # RGB input
-        return torch.cat([input_tensor]*7 + [input_tensor[:,0:1]], dim=1)
+        # Repeat channels to make 22 channels
+        return torch.cat([input_tensor] * 7 + [input_tensor[:, 0:1]], dim=1)
     return input_tensor