yolov5s_beachbot_160

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