Update README.md
Browse files
README.md
CHANGED
|
@@ -10,7 +10,7 @@ base_model:
|
|
| 10 |
# Model Card for Model ID
|
| 11 |
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).
|
| 12 |
|
| 13 |
-
Note that I only use processor
|
| 14 |
|
| 15 |
Another model by directly fine-tuning DETR model can be found at https://huggingface.co/FriedParrot/fish-segmentation-simple
|
| 16 |
|
|
@@ -21,7 +21,7 @@ Tasks :
|
|
| 21 |
- BBoxes prediction, and
|
| 22 |
- Segmentation
|
| 23 |
|
| 24 |
-
> [!warning]
|
| 25 |
> Since this model include customized type, then `AutoModel()` and `AutoConfig()` may fail, but `AutoProcessor()` will work correctly (Since I use a DetrImageProcessor for this)
|
| 26 |
>
|
| 27 |
> 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.
|
|
|
|
| 10 |
# Model Card for Model ID
|
| 11 |
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).
|
| 12 |
|
| 13 |
+
Note that I only use processor of `detr-resnet-50-panotic` and `resnet-50` backbone of the base model, not use transformers. All the model, task heads and loss are self-defined.
|
| 14 |
|
| 15 |
Another model by directly fine-tuning DETR model can be found at https://huggingface.co/FriedParrot/fish-segmentation-simple
|
| 16 |
|
|
|
|
| 21 |
- BBoxes prediction, and
|
| 22 |
- Segmentation
|
| 23 |
|
| 24 |
+
> [!warning]
|
| 25 |
> Since this model include customized type, then `AutoModel()` and `AutoConfig()` may fail, but `AutoProcessor()` will work correctly (Since I use a DetrImageProcessor for this)
|
| 26 |
>
|
| 27 |
> 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.
|