提交 7c1b37d0 编写于 作者: 彭冲 提交者: Waleed

fix inspect_data error by adding gt_class_ids

上级 52bdc66b
......@@ -737,7 +737,7 @@
"source": [
"# Get Next Image\n",
"if random_rois:\n",
" [normalized_images, image_meta, rpn_match, rpn_bbox, gt_boxes, gt_masks, rpn_rois, rois], \\\n",
" [normalized_images, image_meta, rpn_match, rpn_bbox, gt_class_ids, gt_boxes, gt_masks, rpn_rois, rois], \\\n",
" [mrcnn_class_ids, mrcnn_bbox, mrcnn_mask] = next(g)\n",
" \n",
" log(\"rois\", rois)\n",
......@@ -747,6 +747,7 @@
"else:\n",
" [normalized_images, image_meta, rpn_match, rpn_bbox, gt_boxes, gt_masks], _ = next(g)\n",
" \n",
"log(\"gt_class_ids\", gt_class_ids)\n",
"log(\"gt_boxes\", gt_boxes)\n",
"log(\"gt_masks\", gt_masks)\n",
"log(\"rpn_match\", rpn_match, )\n",
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册