diff --git a/.dev/md2yml.py b/.dev/md2yml.py index 4c2e129f27..7fce17a02e 100755 --- a/.dev/md2yml.py +++ b/.dev/md2yml.py @@ -81,6 +81,13 @@ def parse_md(md_file): code_version = None repo_url = None + # To avoid re-counting number of backbone model in OpenMMLab, + # if certain model in configs folder is backbone whose name is already + # recorded in MMClassification, then the `COLLECTION` dict of this model + # in MMSegmentation should be deleted, and `In Collection` in `Models` + # should be set with head or neck of this config file. + is_backbone = None + with open(md_file, 'r') as md: lines = md.readlines() i = 0 @@ -117,9 +124,13 @@ def parse_md(md_file): datasets.append(line[4:]) current_dataset = line[4:] i += 2 + elif line[:15] == '