Update README.md
Browse files
README.md
CHANGED
|
@@ -34,9 +34,9 @@ Pretrained denoising UNet backbones for **Noise2Map: End-to-End Diffusion Model
|
|
| 34 |
## Usage
|
| 35 |
|
| 36 |
```python
|
| 37 |
-
from noise2map import
|
| 38 |
|
| 39 |
-
model =
|
| 40 |
in_channels=6, # 3 for semantic segmentation
|
| 41 |
out_channels=2,
|
| 42 |
img_scale=256,
|
|
|
|
| 34 |
## Usage
|
| 35 |
|
| 36 |
```python
|
| 37 |
+
from noise2map import Noise2Map
|
| 38 |
|
| 39 |
+
model = Noise2Map(
|
| 40 |
in_channels=6, # 3 for semantic segmentation
|
| 41 |
out_channels=2,
|
| 42 |
img_scale=256,
|