Skip to content

Faster-RCNN for The Nature Conservancy Wildlife protection project

Notifications You must be signed in to change notification settings

thomasshupeng/TNC_FRCNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

72 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TNC_FRCNN

Faster-RCNN for The Nature Conservancy Wildlife protection project

Setup

  1. Python 3.5 under Windows and Python 3.5/3.6 under Linux

  2. CNTK 2.6 Python environment CPU-Only:

pip install https://cntk.ai/PythonWheel/CPU-Only/cntk-2.6-cp35-cp35m-win_amd64.whl

or, GPU:

pip install https://cntk.ai/PythonWheel/GPU/cntk_gpu-2.6-cp35-cp35m-win_amd64.whl
  1. Install the following additional packages:
pip install opencv-python easydict pyyaml future

Quick start

  1. If you just want to play with single image prediction Download trained (with 10 classes) model from \shpeng440\TNC_RawData\BU\FasterRCNN_Model Copy the model (faster_rcnn_eval_AlexNet_e2e.model) to TNC_FRCNN\Detection\FasterRCNN\Output folder

  2. If you want to train your own model Go to PretrainedModels folder to download AlexNet_ImageNet_Caffe.model Install VoTT - https://github.com/Microsoft/VoTT#installation Go to DataPrep 2.1 create json for VoTT 2.2 review in VoTT 2.3 export to Faster_RCNN 2.4 create training/test data Go to \Detection\FasterRCNN to train your model Rename Output\faster_rcnn_eval_AlexNet_e2e.model or move to other folder Delete Output\faster_rcnn_eval_AlexNet_e2e.model run 'python run_faster_rcnn.py'

  3. Detect single image Go to \Detection

    python DetectImg.py <image_full_path.jpg>
    

Sub-folder content

Detection

Faster-RCNN training, evaluation and demo.

DataPrep

Scripts for data preparation.

Pretrained Model

The base model for F-RCNN training

DataSets

Contains groups of test/training set. default - BU (data from Beijing University)

[email protected]

About

Faster-RCNN for The Nature Conservancy Wildlife protection project

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published