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

Is PySoloTools suitable for transforming the Multiview SOLO Dataset into COCO Dataset format? #144

Open
Rawdreen opened this issue Aug 24, 2023 · 0 comments

Comments

@Rawdreen
Copy link

In the solo2coco.py script, there is a specific method def _process_instances(...) that solely utilizes the first item within the list, and evidently, this is the limiting factor impeding the conversion process of the multi-camera SOLO Dataset.

@staticmethod
    def _process_instances(
        frame: Frame, idx, output, data_root, solo_kp_map
    ) -> Tuple[Dict, List, List, List]:
        logger.info(f"Processing Frame number: {idx}")
        image_id = idx
        sequence_num = frame.sequence
        rgb_capture = list(
            filter(lambda cap: isinstance(cap, RGBCameraCapture), frame.captures)
        )[0]

        img_record = SOLO2COCOConverter._process_rgb_image(
            image_id, rgb_capture, output, data_root, sequence_num
        )
        (
            ann_record,
            ins_ann_record,
            sem_ann_record,
        ) = SOLO2COCOConverter._process_annotations(
            image_id, rgb_capture, sequence_num, data_root, solo_kp_map
        )

        return img_record, ann_record, ins_ann_record, sem_ann_record

Are there existing methods within PySoloTools to convert a SOLO Dataset with multiple cameras into the COCO format collectively?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant