|
--- |
|
language: en |
|
license: cc-by-nc-4.0 |
|
--- |
|
|
|
# Model Card for yolo5_beachbot_160 |
|
|
|
<!-- Provide a quick summary of what the model is/does. --> |
|
|
|
|
|
|
|
## Model Details |
|
|
|
### Model Description |
|
|
|
<!-- Provide a longer summary of what this model is. --> |
|
|
|
A finetune of Yolo5 intended for identifying beach trash. |
|
|
|
- **Developed by:** Jeffrey Queisser, Christopher Buckley |
|
- **Funded by:** Okinawa Institute of Science and Technology COI Next Grant |
|
- **Model type:** Yolo5 |
|
- **License:** CC BY-NC 4.0 |
|
- **Finetuned from model:** Yolo5 |
|
|
|
### Model Sources |
|
|
|
<!-- Provide the basic links for the model. --> |
|
|
|
- **Repository:** https://github.com/okinawa-ai-beach-robot/yolo5_beachbot_160 |
|
- **Demo:** [Work in Progress] |
|
|
|
## Bias, Risks, and Limitations |
|
|
|
<!-- This section is meant to convey both technical and sociotechnical limitations. --> |
|
|
|
|
|
Note the dataset is small and test set can have quite similar images to the train set. |
|
This was an early version of the dataset so results are not reliable |
|
|
|
|
|
### Recommendations |
|
|
|
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. --> |
|
|
|
|
|
Use later version of the dataset |
|
|
|
|
|
## How to Get Started with the Model |
|
|
|
Use the code below to get started with the model. |
|
|
|
[More Information Needed] |
|
|
|
## Training Details |
|
|
|
### Training Data |
|
|
|
<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. --> |
|
|
|
https://universe.roboflow.com/okinawaaibeachrobot/beach-cleaning-object-detection/dataset/1 |
|
|
|
### Training Procedure |
|
|
|
<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. --> |
|
|
|
#### Preprocessing |
|
|
|
|
|
{'preprocessing': {'auto-orient': True, 'resize': {'width': 1280, 'height': 800, 'format': 'Fill (with center crop) in'}}} |
|
See https://docs.roboflow.com/api-reference/versions/create-a-project-version for more information |
|
|
|
|
|
|
|
#### Training Hyperparameters |
|
|
|
- **Training regime:** |
|
TODO populate with any hyperparameters |
|
<!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision --> |
|
|
|
## Evaluation |
|
|
|
<!-- This section describes the evaluation protocols and provides the results. --> |
|
|
|
### Testing Data, Factors & Metrics |
|
|
|
#### Testing Data |
|
|
|
<!-- This should link to a Dataset Card if possible. --> |
|
|
|
https://universe.roboflow.com/okinawaaibeachrobot/beach-cleaning-object-detection/dataset/1/images?split=test |
|
|
|
#### Metrics |
|
|
|
<!-- These are the evaluation metrics being used, ideally with a description of why. --> |
|
|
|
|
|
TODO populate with any metrics |
|
e.g. loss function |
|
|
|
|
|
### Results |
|
|
|
|
|
TODO populate with any results |
|
|
|
|
|
## Technical Specifications |
|
|
|
## Model Card Authors |
|
|
|
Christopher Buckley |
|
|
|
## Model Card Contact |
|
|
|
https://github.com/okinawa-ai-beach-robot/home/discussions |
