mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

Update get_flops.py (#1371)

Browse Source 
* Update get_flops.py

argparse.ArgumentParser(description='Get the FLOPs of a segmentor')

* Update get_flops.py
This commit is contained in:
wangliujun 2022-03-14 21:02:43 +08:00 committed by GitHub
parent 17b500f9df
commit a3aab38bb4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tools/get_flops.py
View File

@ -8,7 +8,8 @@ from mmseg.models import build_segmentor
def parse_args():
parser = argparse.ArgumentParser(description='Train a segmentor')
parser = argparse.ArgumentParser(
description='Get the FLOPs of a segmentor')
parser.add_argument('config', help='train config file path')
parser.add_argument(
'--shape',