Skip to content

TheodorEmanuelsson/oil_storage-detector

Repository files navigation

Oil Storage Detector

The Data

This dataset is available at Kaggle and contains 98 extract of SPOT imagery at roughly 1.2 meters resolution. Each each image is stored as a JPEG file of size 2560 x 2560 pixels (i.e. 3 kilometers on ground). The locations are selected worldwide. Accompanying the images are annotated polygons which are used in the output layer. There is only one class in the dataset, oil-storage-tank but it is modelled as two classes, one being the storage tank and another being background.

Training sample oil-storage-sample

Training sample with annotated bounding boxes oil-storage-sample_annotated

Since the images are rather large (computationally expensive to train), they are cropped into 512x512 images with 64 pixel overlappes. This resulted in 468 cropped training images and 120 cropped valiation images. There are also 6 test images 5 which are not annotated but could be used to visually test a model on new - unseen before - images.

The Model

The YOLOv5 model is used with pretrained weights which are updates during further training. Due to the low number of images in the dataset, transfer learning was required to get a well performing model.

Training

The model was trained for 150 epochs with a batch size of 32 using SGD

Precision-Recall curve

PR_curve

Training metrics over epochs

results

Prediction

Using a confidence cutoff of 0.65, the following results was achieved.

Snowy image with few storage tanks

67f7c7ad-11a1-4c7f-9f2a-da7ef50bfdd8

Image with lots of storage tanks

605ffac0-69d5-4748-92c2-48d43f51afc1