[Fix] fix model_zoo script

.pre-commit-config.yaml

@@ -64,3 +64,13 @@ repos:
       - id: check-copyright
         args: ["demo", "mmedit", "tests", "tools"]
       - id: remove-improper-eol-in-cn-docs
+  - repo: local
+    hooks:
+      - id: update-model-zoo
+        name: update-model-zoo
+        description: update model-zoo.yml
+        entry: docs/en/.dev_scripts/update_model_zoo.py
+        additional_dependencies: [mmengine, modelindex, tqdm, rich]
+        language: python
+        files: ^configs/.*\.md$
+        require_serial: true

docs/en/.dev_scripts/update_model_zoo.py

@@ -1,3 +1,4 @@
+#!/usr/bin/env python
 import os
 from glob import glob
 from os import path as osp
@@ -52,9 +53,9 @@ def update_model_zoo():
     papers = set()
     checkpoints = set()
     for m in full_models:
-        papers.update(m.paper)
+        papers.add(m.paper['Title'])
         if m.weights is not None and m.weights.startswith('https:'):
-            checkpoints.update(m.weights)
+            checkpoints.add(m.weights)
     task_desc = '\n'.join([
         f"  - [{t}]({t.replace('-', '_').replace(' ', '_')}.md)"
         for t in list(tasks.keys())
@@ -90,12 +91,12 @@ def update_model_zoo():
         checkpoints = set()
         for m in models:
             if m.weights is not None and m.weights.startswith('https:'):
-                checkpoints.update(m.weights)
+                checkpoints.add(m.weights)
         collections = set([m.in_collection for m in models])
 
         papers = set()
         for m in models:
-            papers.update(m.paper)
+            papers.add(m.paper['Title'])
 
         content = ''
         readme = set()