sync

.gitattributes

@@ -1,35 +1,34 @@
 *.7z filter=lfs diff=lfs merge=lfs -text
 *.arrow filter=lfs diff=lfs merge=lfs -text
 *.bin filter=lfs diff=lfs merge=lfs -text
+*.bin.* filter=lfs diff=lfs merge=lfs -text
 *.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
 *.ftz filter=lfs diff=lfs merge=lfs -text
 *.gz filter=lfs diff=lfs merge=lfs -text
 *.h5 filter=lfs diff=lfs merge=lfs -text
 *.joblib filter=lfs diff=lfs merge=lfs -text
 *.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
 *.model filter=lfs diff=lfs merge=lfs -text
 *.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
 *.onnx filter=lfs diff=lfs merge=lfs -text
 *.ot filter=lfs diff=lfs merge=lfs -text
 *.parquet filter=lfs diff=lfs merge=lfs -text
 *.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
 *.pt filter=lfs diff=lfs merge=lfs -text
 *.pth filter=lfs diff=lfs merge=lfs -text
 *.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
 saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
 *.tflite filter=lfs diff=lfs merge=lfs -text
 *.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
 *.xz filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*.tfevents* filter=lfs diff=lfs merge=lfs -text
+*.db* filter=lfs diff=lfs merge=lfs -text
+*.ark* filter=lfs diff=lfs merge=lfs -text
+**/*ckpt*data* filter=lfs diff=lfs merge=lfs -text
+**/*ckpt*.meta filter=lfs diff=lfs merge=lfs -text
+**/*ckpt*.index filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,2 @@
+save.pt
+rename.sh

README.md

@@ -1,10 +1,10 @@
 ---
 title: HEp2
-emoji: ⚡
+emoji: 🧬🧫
 colorFrom: blue
 colorTo: pink
 sdk: gradio
-sdk_version: 4.12.0
+sdk_version: 4.39.0
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -0,0 +1,67 @@
+import os
+import torch
+import gradio as gr
+from PIL import Image
+from torchvision.transforms import transforms
+from modelscope import snapshot_download
+
+MODEL_DIR = snapshot_download("MuGeminorum/HEp2", cache_dir="./__pycache__")
+TRANSLATE = {
+    "Centromere": "着丝粒 Centromere",
+    "Golgi": "高尔基体 Golgi",
+    "Homogeneous": "同质 Homogeneous",
+    "NuMem": "记忆体 NuMem",
+    "Nucleolar": "核仁 Nucleolar",
+    "Speckled": "斑核 Speckled",
+}
+CLASSES = list(TRANSLATE.keys())
+
+
+def embeding(img_path: str):
+    compose = transforms.Compose(
+        [
+            transforms.Resize(224),
+            transforms.CenterCrop(224),
+            transforms.RandomAffine(5),
+            transforms.ToTensor(),
+            transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
+        ]
+    )
+    img = Image.open(img_path).convert("RGB")
+    return compose(img)
+
+
+def infer(target: str):
+    model = torch.load(f"{MODEL_DIR}/save.pt", map_location=torch.device("cpu"))
+    if not target:
+        return None, "请上传细胞图片 Please upload a cell picture!"
+
+    torch.cuda.empty_cache()
+    input: torch.Tensor = embeding(target)
+    output: torch.Tensor = model(input.unsqueeze(0))
+    predict = torch.max(output.data, 1)[1]
+    return os.path.basename(target), TRANSLATE[CLASSES[predict]]
+
+
+if __name__ == "__main__":
+    example_imgs = []
+    for cls in CLASSES:
+        example_imgs.append(f"{MODEL_DIR}/examples/{cls}.png")
+
+    with gr.Blocks() as demo:
+        gr.Interface(
+            fn=infer,
+            inputs=gr.Image(
+                type="filepath", label="上传细胞图像 Upload a cell picture"
+            ),
+            outputs=[
+                gr.Textbox(label="图片名 Picture name", show_copy_button=True),
+                gr.Textbox(label="识别结果 Recognition result", show_copy_button=True),
+            ],
+            title="请上传 PNG 格式的 HEp2 细胞图片<br>It is recommended to upload HEp2 cell images in PNG format.",
+            examples=example_imgs,
+            allow_flagging="never",
+            cache_examples=False,
+        )
+
+    demo.launch()

requirements.txt

@@ -0,0 +1,7 @@
+torch
+gradio
+pillow
+torchvision
+tqdm
+requests
+modelscope==1.15