5 lines
180 B
Python
5 lines
180 B
Python
import torch
|
|
from cloud_removal import DeepInpaintingStrategy
|
|
cloud_remover = DeepInpaintingStrategy()
|
|
print("Model channels:", list(cloud_remover.model.parameters())[0].shape[1])
|