Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Feat/2361 segmentation mask #2426

Merged
merged 6 commits into from
Nov 11, 2024
Merged
Prev Previous commit
Next Next commit
2361-SegmentationMask removed unnecessary serialize/deserialize
anthonytorlucci committed Oct 30, 2024

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
commit ca13b9f7468d2e635716275ee004773ed67857ce
2 changes: 1 addition & 1 deletion crates/burn-dataset/src/vision/image_folder.rs
Original file line number Diff line number Diff line change
@@ -73,7 +73,7 @@ pub enum Annotation {
/// Segmentation mask annotation.
/// For semantic segmentation, a mask has a single channel (C = 1).
/// For instance segmentation, there may be multiple masks per image (C >= 1).
#[derive(Deserialize, Serialize, Debug, Clone, PartialEq)]
#[derive(Debug, Clone, PartialEq)]
pub struct SegmentationMask {
/// Segmentation mask.
pub mask: Vec<usize>,