Skip to content

Commit

Permalink
Fix: convert images to RGB (#69)
Browse files Browse the repository at this point in the history
* Fix: convert images to RGB

* Change the source branch to install

* [Automated] Updated coverage badge

---------

Co-authored-by: GitHub Actions <actions@github.com>
  • Loading branch information
HonzaCuhel and actions-user authored Oct 30, 2024
1 parent 1914f7d commit 72f2aa4
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 5 deletions.
4 changes: 3 additions & 1 deletion datadreamer/pipelines/generate_dataset_from_scratch.py
Original file line number Diff line number Diff line change
Expand Up @@ -552,7 +552,9 @@ def read_image_batch(image_batch, batch_num, batch_size):
batch_image_paths.append(image_path)

else:
images = [Image.open(image_path) for image_path in image_batch]
images = [
Image.open(image_path).convert("RGB") for image_path in image_batch
]
batch_image_paths = image_batch
return images, batch_image_paths

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
},
"outputs": [],
"source": [
"!pip install -q datadreamer@git+https://github.com/luxonis/datadreamer@dev"
"!pip install -q datadreamer"
]
},
{
Expand Down
6 changes: 3 additions & 3 deletions media/coverage_badge.svg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit 72f2aa4

Please sign in to comment.