Skip to content

SpaceParkLeicester/yolo_otd

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

29 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

YOLO-Airbus Oil tank detection

Modular implementation of YOLOX

Installation - Requirements

Note: Both CUDA-10.1 and CUDA-10.2 should be added to /usr/local/, and add below line to the .bashrc

export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

Make sure to run environment.yml to install all necessary packages, and the YOLOX detection is adapted from the given link, make sure to clone the repo and install it as editable.

and

Download https://github.com/Megvii-BaseDetection/YOLOX/releases/download/0.1.1rc0/yolox_s.pth

Configuration

Make sure to run the coco and configuration files as explained in the instructions, especially adding the "oil" in the COCO class pre-defined coco classes

Instructions of usage

  • Add YOLOX as env with path to YOLOX detection
  • Add AIRBUS_SPOT as env with path to Airbus imagery folder. src/cfg/splits.py-->src/coco/tiles.py--->src/coco/oil_coco.py--->src/coco/annots.py

About

ML models for Object Detection on Airbus data

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published