Commit
·
6e2081d
1
Parent(s):
c2bfa14
Fixed
Browse files- .gitignore +1 -0
- inference.py +3 -0
.gitignore
CHANGED
|
@@ -4,3 +4,4 @@ data
|
|
| 4 |
samples/*
|
| 5 |
!samples/.gitkeep
|
| 6 |
model_cache
|
|
|
|
|
|
| 4 |
samples/*
|
| 5 |
!samples/.gitkeep
|
| 6 |
model_cache
|
| 7 |
+
image_processor_cache
|
inference.py
CHANGED
|
@@ -15,6 +15,9 @@ image_processor = AutoImageProcessor.from_pretrained(
|
|
| 15 |
|
| 16 |
|
| 17 |
def inference(rgb: Image.Image, depth: Image.Image) -> Image.Image:
|
|
|
|
|
|
|
|
|
|
| 18 |
preprocessed_sample: Dict[str, Tensor] = image_processor.preprocess(
|
| 19 |
{
|
| 20 |
"rgb": rgb,
|
|
|
|
| 15 |
|
| 16 |
|
| 17 |
def inference(rgb: Image.Image, depth: Image.Image) -> Image.Image:
|
| 18 |
+
rgb = rgb.convert(mode="RGB")
|
| 19 |
+
depth = depth.convert(mode="L")
|
| 20 |
+
|
| 21 |
preprocessed_sample: Dict[str, Tensor] = image_processor.preprocess(
|
| 22 |
{
|
| 23 |
"rgb": rgb,
|