Views
No views yet
1from ultralytics import YOLO
2from huggingface_hub import hf_hub_download
3
4# Download weights from Hugging Face
5model_path = hf_hub_download(repo_id="vitala100/brick-kiln-yolo-detector", filename="brick_kilns_finder_yolo_model.pt")
6
7# Load YOLO model
8model = YOLO(model_path)
9
10# Run detection on satellite/aerial image
11results = model("satellite_image.jpg")
12
13# Display results
14results[0].show()