Update README.md
Browse files
README.md
CHANGED
@@ -83,7 +83,7 @@ image2 = Image.open(file2)
|
|
83 |
|
84 |
|
85 |
|
86 |
-
foregrounds = model.inference([image1, image2]) # We
|
87 |
foregrounds[0].save("./foreground1.png")
|
88 |
foregrounds[1].save("./foreground2.png")
|
89 |
|
|
|
83 |
|
84 |
|
85 |
|
86 |
+
foregrounds = model.inference([image1, image2]) # We recommend that the batch size not exceed 3 for consumer GPUs as there are minimal inference gains due to our custom batch processing for the MVANet decoding steps.
|
87 |
foregrounds[0].save("./foreground1.png")
|
88 |
foregrounds[1].save("./foreground2.png")
|
89 |
|