DeepGeo Extension :: YOLO

• Easy Deep Learning

• Copyright (c) 2019 InfoLab (Donggun LEE)

• How to install

  pip install deepgeo_ext_yolo
  

  • other version

    # 0.0.1
    pip install deepgeo_ext_yolo==0.0.1
    

  • requirement
    • Python 3.6

    pip install deepgeo opencv-python numpy
    

• How to use

  import deepgeo
  
  engine = deepgeo.Engine()
  engine.add_model('maskrcnn_yolo','yolo','D:/default_config.json')
  
  image = deepgeo.Image("image.jpg","D:/Project")
  image = engine.detect('maskrcnn_yolo', image)[0]
  image.draw_annotations(image.get_annotation())
  image.save("D:/","test","PNG")
  

• default_config.json

    {
       "VERSION" = ""
       "MODEL_FILE_NAME" : "road-damage.weights",
       "MODEL_PATH" : "model",
       "CLASSES" : "road-damage-class.txt",
       "CONFIG" : "road-damage.cfg",
       "MEMO" : "road damage detect"
    }