Skip to content

Commit

Permalink
target class type
Browse files Browse the repository at this point in the history
  • Loading branch information
MeinLieben committed Oct 18, 2024
1 parent 25d0164 commit 4e9c8d3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion demo/inference_img.py
Original file line number Diff line number Diff line change
@@ -259,7 +259,7 @@ def _inference_one(img, maskdts, curr_mask, target_class, out_writer): # to save
parser.add_argument('--video', required=True, help='path to the rgb video')
parser.add_argument('--masks_video', required=True, help='path to the video with masks')
parser.add_argument('--out_video', required=True, help='path to the output video with masks')
parser.add_argument('--target_class', required=False, default=None, help='mark of the target semantic class')
parser.add_argument('--target_class', required=False, default=None, type=int, help='mark of the target semantic class')

args = parser.parse_args()

0 comments on commit 4e9c8d3

Please sign in to comment.