Skip to content

[ISSUE]: Several issues #35

@haavarpb

Description

@haavarpb

Description

Hi,

I am a PhD student at NMBU, trying to use your repository for a baseline comparison with other segmentation methods. I have several issues:

  • Pretrained models don't have any accompanying instructions on configuration. Currently, you set the num_classes from what is set under training.num_classes. This, I expect, would change based on the model used.

    cfg.MODEL.WEIGHTS = os.path.join(self.model_file)
    cfg.MODEL.ROI_HEADS.NUM_CLASSES = self.num_classes

    Loading a different model e.g. aoc_strawberry_class_fruit.pth results in misconfigured model.

  • Missing metadata catalogs for other models.
    The accompanying metadata pickle file contain two classes, ripe and unripe. So I guess this metadata file is only valid for the model for ripeness. Where are the ones for the other models? I feel there is missing description for each available model and how they are trained.

  • It is not described anywhere what backbone should/could be used. Currently you have:

    config_file: 'COCO-InstanceSegmentation/mask_rcnn_R_101_FPN_3x.yaml'

Does this insinuate freedom to choose otherwise, or should this be static?

  • Broken LearnerUtils.download()
    Downloading datasets and models using LearnerUtils is not possible without configuring requests to not verify SSL certificates (verify=False)
    with requests.get(url, stream=True) as r:

Otherwise, thanks for putting this repo out there!

Metadata

Metadata

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions