kardosdrur's picture
Update README.md
7bc4c32 verified
metadata
language:
  - en
tags:
  - yolo
  - object-detection
library_name: burial_mounds
license: cc-by-nc-4.0

kardosdrur/burial-mounds-yolov8m-xview

This repository contains a YOLO model that has been finetuned by the burial_mounds Python package on the Mounds dataset.

The model is for academic use only, commercial use is prohibited due to restrictions imposed by the training datasets.

Usage

# pip install burial_mounds

from burial_mounds.model import MoundDetector

model = MoundDetector.load_from_hub("kardosdrur/burial-mounds-yolov8m-xview")

# Find bounding polygons
bounding_polygons = model.detect_mounds("some_satellite_image.png")
for polygon in bounding_polygons:
    print(polygon)

# Annotate satellite images
annotated_image = model.annotate_image("some_satellite_image.png")
annotated_image.show()

For a more detailed guide consult the YOLOv8 documentation or our documentation.