Skip to content

Commit

Permalink
Update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
TheDevilWillBeBee committed Jun 2, 2024
1 parent a65670b commit 3fd4b10
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -104,7 +104,7 @@ If you face out of memory issues, you can reduce the `num_views` or the `batch_s

```yaml
train:
batch_size: 1 # Batch size
batch_size: 1
camera:
num_views: 6
```
Expand Down
3 changes: 2 additions & 1 deletion utils/render.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,6 +123,7 @@ def to_pil(rendered_features, target_channels=(0, 3),
:return: A PIL Image showing the rendered images.
"""
# @ TODO This function is not consistent with the meshnca.render_channels
assert rendered_features.dim() == 5, "The input tensor should have 5 dimensions"

if not isinstance(target_channels, dict):
Expand All @@ -143,7 +144,7 @@ def to_pil(rendered_features, target_channels=(0, 3),
) # [batch_size*height, num_views*width, num_features]

image_list = []
for key, channels in target_channels.items():
for key, channels in sorted(target_channels.items()):
image = features[..., range(*channels)]
if image.shape[-1] == 1:
image = np.repeat(image, 3, axis=-1)
Expand Down

0 comments on commit 3fd4b10

Please sign in to comment.