Commit d2d8d9c0 authored by suilin0432's avatar suilin0432
Browse files

updage wsis visualization

parent eb50d0b3
......@@ -733,6 +733,7 @@ class StandardROIHeads(ROIHeads):
# Usually the original proposals used by the box head are used by the mask, keypoint
# heads. But when `self.train_on_pred_boxes is True`, proposals will contain boxes
# predicted by the box head.
# 在训练的时候调用 forward_mask, 在测试的时候回调用 forward_with_given_boxes
losses.update(self._forward_mask(features, proposals))
losses.update(self._forward_keypoint(features, proposals))
return proposals, losses
CUDA_VISIBLE_DEVICES=4,5,6,7 python tools/ --input output/wsis/voc_plus_1_4w_008/inference/coco_instances_results.json --output output/wsis/voc_plus_1_4w_008/visualization --dataset voc_2012_seg_val --conf-threshold 0.5
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment