Skip to content

Commit

Permalink
add coco param
Browse files Browse the repository at this point in the history
  • Loading branch information
ijkguo committed Jun 16, 2018
1 parent 6d74943 commit a02850b
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions sym_demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,6 +99,7 @@ def get_voc_names(args):

def get_coco_names(args):
from symimdb.coco import coco
args.rpn_anchor_scales = (2, 4, 8, 16, 32)
args.rcnn_num_classes = len(coco.classes)
return coco.classes

Expand Down

0 comments on commit a02850b

Please sign in to comment.