refactor: reorganize project structure by moving core modules and update import paths in API server

2026-07-18 01:24:30 +07:00
parent abab846884
commit a82b2f6fa5
155 changed files with 25 additions and 370 deletions
@@ -0,0 +1,7 @@
+import torch
+checkpoint = torch.load('cloud_removal_model/cloud_removal_unet_best.pth', map_location='cpu')
+print(checkpoint.keys())
+print("in_channels in checkpoint:", 'in_channels' in checkpoint)
+if 'in_channels' in checkpoint:
+    print(checkpoint['in_channels'])
+print("Shape of inc.double_conv.0.weight:", checkpoint['model_state_dict']['inc.double_conv.0.weight'].shape)