FriedParrot commited on
Commit
17a1fdd
·
verified ·
1 Parent(s): f03bea8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 and `resnet-50` backbone of `resnet-50-panotic` 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,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.