Modular implementation of YOLOX
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
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
- 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