Skip to content

ShawnXuan/flowvision_detection_example

Repository files navigation

OneFlow Flowvision Detection Examples.

Training with coco dataset

./train.sh [/path/to/coco/dataset] [#GPUS] [model_name] [batch_size]

Evaluation with coco dataset

./eval.sh [/path/to/coco/dataset] [#GPUS] [model_name] [batch_size]

model_names = [ssd300_vgg16, fasterrcnn_resnet50_fpn, maskrcnn_resnet50_fpn]

Download coco2017 from links below

unzipping

unzip *.zip

pretraind models

Pretrained model will be automatically downloaded to $HOME/.oneflow/flowvision_cache directory.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published