README.md

metadata

library_name: transformers
license: creativeml-openrail-m
language:
  - en
base_model:
  - facebook/detr-resnet-50-panoptic

Model Card for Model ID

This is a Model using ResNet-50 backbone and customized Multi-task Head and loss to make classification, boundary box prediction and segmentation (24.7M parameters).

Note that I only use processor and resnet-50 backbone of resnet-50-panotic model, not use transformers. All the model, task heads and loss are self-defined.

Another model by directly fine-tuning DETR model can be found at https://huggingface.co/FriedParrot/fish-segmentation-simple

This model use kaggle dataset A Large Scale Fish Dataset as dataset for training.

Tasks :

• Classification
• BBoxes prediction, and
• Segmentation

Since this model include customized type, then AutoModel() and AutoConfig() may fail, but AutoProcessor() will work correctly (Since I use a DetrImageProcessor for this)

If you want using this model, you can go to my source code below and look for FishSegmentationModel and FishSegmentModelConfig for load these models correctly.

Model Sources

For source code : check my github

Results and test

check my github. Note this may not be updated in time.