nielsr HF Staff commited on
Commit
f360d76
·
verified ·
1 Parent(s): 54c80ec

Improve model card

Browse files

Hi! I'm Niels, part of the community science team at Hugging Face. I'm opening this PR to improve the model card for EdgeCrafter.

The changes include:
- Adding the `pipeline_tag: object-detection` to the YAML metadata for better discoverability.
- Specifying the `license: apache-2.0` based on the project's repository.
- Adding links to the paper, project page, and source code.
- Including a summary and sample usage instructions derived from the official GitHub README.

Files changed (1) hide show
  1. README.md +49 -4
README.md CHANGED
@@ -1,10 +1,55 @@
1
  ---
 
 
2
  tags:
3
  - model_hub_mixin
4
  - pytorch_model_hub_mixin
5
  ---
6
 
7
- This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
8
- - Code: https://github.com/Intellindust-AI-Lab/EdgeCrafter
9
- - Paper: https://arxiv.org/abs/2603.18739
10
- - Docs: [More Information Needed]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ license: apache-2.0
3
+ pipeline_tag: object-detection
4
  tags:
5
  - model_hub_mixin
6
  - pytorch_model_hub_mixin
7
  ---
8
 
9
+ # EdgeCrafter: Compact ViTs for Edge Dense Prediction via Task-Specialized Distillation
10
+
11
+ EdgeCrafter is a unified compact Vision Transformer (ViT) framework designed for high-performance dense prediction (object detection, instance segmentation, and pose estimation) on resource-constrained edge devices.
12
+
13
+ - **Paper:** [EdgeCrafter: Compact ViTs for Edge Dense Prediction via Task-Specialized Distillation](https://huggingface.co/papers/2603.18739)
14
+ - **Project Page:** [https://intellindust-ai-lab.github.io/projects/EdgeCrafter/](https://intellindust-ai-lab.github.io/projects/EdgeCrafter/)
15
+ - **Repository:** [https://github.com/Intellindust-AI-Lab/EdgeCrafter](https://github.com/Intellindust-AI-Lab/EdgeCrafter)
16
+
17
+ ## Introduction
18
+ Deploying high-performance dense prediction models on edge devices is challenging due to strict computation and memory limits. EdgeCrafter introduces a framework centered on **ECDet**, a detection model built from a distilled compact backbone and an edge-friendly encoder-decoder design. On the COCO dataset, ECDet-S achieves 51.7 AP with fewer than 10M parameters.
19
+
20
+ ## Quick Start (Inference)
21
+ The easiest way to test EdgeCrafter is to run inference on a sample image using the provided tools in the official repository.
22
+
23
+ ### Installation
24
+ ```bash
25
+ # Create conda environment
26
+ conda create -n ec python=3.11 -y
27
+ conda activate ec
28
+
29
+ # Install dependencies
30
+ pip install -r requirements.txt
31
+ ```
32
+
33
+ ### Inference
34
+ ```bash
35
+ # 1. Download a pre-trained model (e.g., ECDet-L)
36
+ cd ecdetseg
37
+ wget https://github.com/capsule2077/edgecrafter/releases/download/edgecrafterv1/ecdet_l.pth
38
+
39
+ # 2. Run PyTorch inference
40
+ # Make sure to replace `path/to/your/image.jpg` with an actual image path
41
+ python tools/inference/torch_inf.py -c configs/ecdet/ecdet_l.yml -r ecdet_l.pth -i path/to/your/image.jpg
42
+ ```
43
+
44
+ ## Citation
45
+ ```bibtex
46
+ @article{liu2026edgecrafter,
47
+ title={EdgeCrafter: Compact ViTs for Edge Dense Prediction via Task-Specialized Distillation},
48
+ author={Liu, Longfei and Hou, Yongjie and Li, Yang and Wang, Qirui and Sha, Youyang and Yu, Yongjun and Wang, Yinzhi and Ru, Peizhe and Yu, Xuanlong and Shen, Xi},
49
+ journal={arXiv},
50
+ year={2026}
51
+ }
52
+ ```
53
+
54
+ ---
55
+ This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration.