(self, device)
| 706 | |
| 707 | class Image2Normal: |
| 708 | def __init__(self, device): |
| 709 | print("Initializing Image2Normal") |
| 710 | self.depth_estimator = pipeline("depth-estimation", model="Intel/dpt-hybrid-midas") |
| 711 | self.bg_threhold = 0.4 |
| 712 | |
| 713 | @prompts(name="Predict Normal Map On Image", |
| 714 | description="useful when you want to detect norm map of the image. " |
nothing calls this directly
no outgoing calls
no test coverage detected