mirrors
/
SegFormer
mirror of https://github.com/NVlabs/SegFormer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge 22f1b51cae into 65fa8cfa9b

pull/143/merge
LWShowTime 2023-12-19 16:38:45 +08:00 committed by GitHub
parent 65fa8cfa9b 22f1b51cae
commit 842e7736d7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
demo/image_demo.py
View File

@ -6,9 +6,9 @@ from mmseg.core.evaluation import get_palette
def main():
parser = ArgumentParser()
parser.add_argument('img', help='Image file')
parser.add_argument('config', help='Config file')
parser.add_argument('checkpoint', help='Checkpoint file')
parser.add_argument('--img', help='Image file', default="XX")
parser.add_argument('--config', help='Config file', default="XX")
parser.add_argument('--checkpoint', help='Checkpoint file', default="XX")
parser.add_argument(
'--device', default='cuda:0', help='Device used for inference')
parser.add_argument(