11 lines
334 B
Python
11 lines
334 B
Python
from cloud_removal import DeepInpaintingStrategy
|
|
import torch
|
|
import numpy as np
|
|
|
|
cr = DeepInpaintingStrategy(model_path="cloud_removal_model/cloud_removal_unet_best.pth")
|
|
if cr.model is not None:
|
|
expected = list(cr.model.parameters())[0].shape[1]
|
|
print("Expected channels:", expected)
|
|
else:
|
|
print("Failed to load model")
|