Skip to content

Integrate segmentation pipeline into custom dataloader #19

@suryajasper

Description

@suryajasper

Modify the load_image function in the dataloader to segment and crop the relevant portion of the image using the segmentation pipeline

load_image:

def _load_image(self, image_path):
image = Image.open(image_path).convert('RGB')
if self.transform is not None:
image = self.transform(image)
return image

segmentation pipeline: https://github.com/suryajasper/SynthMining/blob/main/synthesis_pipeline/object_segmentation/object_segmentation.py

Metadata

Metadata

Assignees

Labels

No labels
No labels

Projects

Status

In Progress

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions