README.md

---
license: mit
datasets:
- EnmmmmOvO/SeaTurtleID2022
language:
- en
base_model:
- torchvision.models.segmentation.fcn_resnet101
- segmentation_models_pytorch.fpn
---
# Model Card for Model ID

This repository contains FCN and FPN models trained using the split_open strategy from [SeaTurtleID2022](https://www.kaggle.com/datasets/wildlifedatasets/seaturtleid2022). The models have been uploaded to facilitate evaluation and review by the supervisor.

## Model Details

### Model Description

- **Developed by:** Jinghan Wang
- **Language(s) (NLP):** English
- **License:** MIT

### Model Sources [optional]

- **FPN**: [Segmentation Models Pytorch](https://segmentation-modelspytorch.readthedocs.io/en/latest/docs/api.html#fpn)
- **FCN:** [Pytorch](https://pytorch.org/vision/main/models/generated/torchvision.models.segmentation.fcn_resnet101.html#fcn-resnet101)
- 
## Uses

These models have been uploaded to facilitate review and evaluation for the COMP9517 Computer Vision project assignment at the University of New South Wales, T3 2024 semester.

## Training Details

### Training Data

[SeaTurtleID2022](https://www.kaggle.com/datasets/wildlifedatasets/seaturtleid2022)
`split_open: 'train'`

### Valid Data

[SeaTurtleID2022](https://www.kaggle.com/datasets/wildlifedatasets/seaturtleid2022)
`split_open: 'valid'`


### Testing Data, Factors & Metrics

#### Testing Data

[SeaTurtleID2022](https://www.kaggle.com/datasets/wildlifedatasets/seaturtleid2022)
`split_open: 'test'`


## Evaluation
```
FCN (Resnet101)
  Mean IoU: 0.9039
  Mean Accuracy: 0.9458

  Mean IoU of backdrop: 0.9932
  Mean Accuracy of backdrop: 0.9963

  Mean IoU of turtle: 0.9225
  Mean Accuracy of turtle: 0.9713

  Mean IoU of flipper: 0.8351
  Mean Accuracy of flipper: 0.8940

  Mean IoU of head: 0.8649
  Mean Accuracy of head: 0.9215
```

```
FPN (Resnet152):
  Mean IoU: 0.9042
  Mean Accuracy: 0.9440

  Mean IoU of backdrop: 0.9933
  Mean Accuracy of backdrop: 0.9966

  Mean IoU of turtle: 0.9242
  Mean Accuracy of turtle: 0.9708

  Mean IoU of flipper: 0.8317
  Mean Accuracy of flipper: 0.8899

  Mean IoU of head: 0.8677
  Mean Accuracy of head: 0.9187
```

## Environmental Impact

- **Cloud Provider:** [Colab](https://colab.research.google.com/)