项目作者: miamor

项目描述 :
Tools for some model implemented in caffe framework
高级语言: Python
项目地址: git://github.com/miamor/CF_tools.git
创建时间: 2018-03-09T02:38:36Z
项目社区:https://github.com/miamor/CF_tools

开源协议:

下载


CF_tools

Running caffe with SSD, FSSD, YOLO implemented
Install caffe and its python module here

Simple run

./tool.sh

Get to more details

Prepare data before train:

  1. ./prepare_data.sh \
  2. -g <dataset_gen_dir> \
  3. -d <dataset dir> \
  4. -m <model>

Run tool

  1. ./run_tool.sh \
  2. -t <train|detect|evaluate> \
  3. -g <Generation path> \
  4. -m <Model (ssd|fssd|yolo)> \
  5. -w <weight_file> \
  6. -s <img_size> \
  7. -u <gpu_id>

Train with your own dataset