/MaskYolo_Caffe

YOLO V2 & V3 , YOLO Combined with RCNN and MaskRCNN

Primary LanguageC++

YOLO Caffe version with MaskRCNN

caffe-maskyolo

  1. What I add in this version of caffe?
  • detection lmdb, mask lmdb, keypoint lmdb prepare
  • yolo v2 (RegionLossLayer) and v3 (YoloLossLayer) are supported
  • yolo combined with maskrcnn
  1. Compile it
cd caffe-maskyolo
cp Makefile.config.example Makefile.config
make -j

Object Detection with YOLO

support to use yolo v2 or v3 to detect objects in images

  1. examples of detection on KITTI

  2. prepare lmdb for object detection

cd ROOT
sh ./maskyolo/scripts/convert_detection.sh  #generate lmdb for detection
cd ./maskyolo/models/mobilenetv2-yolo/
nohup sh yolo_train.sh > train.log &
tail -f train.log
  1. objection demo
comming soon

Instance mask and Keypoint recognization with MaskRCNN combined YOLO

Use yolo results as input to feed to roi_ppooing or roi_alignment layer

  1. examples of mask & keypoints on COCO

  2. prepare lmdb for mask regression

coming soon
  1. yolo with mask demo
comming soon

Reference

You Only Look Once: Unified, Real-Time Object detection http://arxiv.org/abs/1506.02640

YOLO9000: Better, Faster, Stronger https://arxiv.org/abs/1612.08242

YOLOv3: An Incremental Improvement https://pjreddie.com/media/files/papers/YOLOv3.pdf

Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks

Mask R-CNN