seg_every_thing icon indicating copy to clipboard operation
seg_every_thing copied to clipboard

train on own dataset

Open wangshuai09 opened this issue 7 years ago • 3 comments

I convert my dataset to the coco format,the dataset has no segmentation,so i set the "segmentation" =[] in annotations,and set the cfg.TRAIN.MRCNN_LABELS_TO_KEEP = ( ),i also convert the pretrained coco MXRCNN model to my dataset followed the convert_coco_model_cityscapes.py

finally, i got the error: Exception encountered running PythonOp function: ValueError: min() arg is an empty sequence

Is the cfg.TRAIN.MRCNN_LABELS_TO_KEEP =( ) means that segmention would not be used? how should i train bbox branch on my dataset which without segmentation from the coco pretrained MXRCNN model?

wangshuai09 avatar Jul 20 '18 03:07 wangshuai09

@xiaozr @ronghanghu I met same problem, have you solved it?

jinxuan777 avatar Aug 01 '18 12:08 jinxuan777

i just set the "segmentation" = [[1,1,1,1,1,1]] ,and during the training ,the mask loss is 0.0

wangshuai09 avatar Aug 03 '18 02:08 wangshuai09

@xiaozr May I ask you that how do you install seg_every_thing? Looking forward to your reply!

lssily avatar Jan 23 '19 03:01 lssily