Update README.md
Browse files
README.md
CHANGED
@@ -1,3 +1,70 @@
|
|
1 |
-
---
|
2 |
-
license:
|
3 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
---
|
4 |
+
# DiffusionEdge Model Weights
|
5 |
+
|
6 |
+
**Important Notice: These model weights are NOT owned by the uploader of this repository. This is a redistribution of the original DiffusionEdge model weights for convenience.**
|
7 |
+
|
8 |
+
## Model Overview
|
9 |
+
|
10 |
+
This repository contains the pre-trained model weights for DiffusionEdge, a diffusion probabilistic model for crisp edge detection. These weights are redistributed here to support the MooneyMaker project, which uses DiffusionEdge for automatic creation of Mooney images.
|
11 |
+
|
12 |
+
## Original Work Attribution
|
13 |
+
|
14 |
+
- **Original Repository**: [GuHuangAI/DiffusionEdge](https://github.com/GuHuangAI/DiffusionEdge)
|
15 |
+
- **Original Authors**: Yunfan Ye, Yuhang Huang, Renjiao Yi, Zhiping Cai, Kai Xu
|
16 |
+
- **Paper**: ["DiffusionEdge: Diffusion Probabilistic Model for Crisp Edge Detection"](https://arxiv.org/abs/2401.02032)
|
17 |
+
|
18 |
+
## License and Legal
|
19 |
+
|
20 |
+
These model weights are redistributed under the same LICENSE (Apache-2.0) as the original DiffusionEdge repository. The original code and model weights were made available by the authors for research and development purposes.
|
21 |
+
|
22 |
+
**Users of these weights must comply with all terms and conditions of the original license and repository.**
|
23 |
+
|
24 |
+
For any legal questions regarding the use of these weights, please refer to the original repository or contact the original authors.
|
25 |
+
|
26 |
+
## Model Details
|
27 |
+
|
28 |
+
## Usage in MooneyMaker
|
29 |
+
|
30 |
+
These weights are specifically used in the MooneyMaker project for edge detection as part of the Mooney image generation process. The edge detection output helps to select Mooney images whose edge maps are either similar to dissimilar to those of the template image.
|
31 |
+
If you use the MooneyMaker package, the weights should be downloaded automatically. If not please place them in `mooney_maker/DiffusionEdge`.
|
32 |
+
|
33 |
+
## Original Citation
|
34 |
+
|
35 |
+
If you use these weights in your research or project, please cite the original paper:
|
36 |
+
|
37 |
+
```bibtex
|
38 |
+
@inproceedings{ye2024diffusionedge,
|
39 |
+
title={DiffusionEdge: Diffusion Probabilistic Model for Crisp Edge Detection},
|
40 |
+
author={Yunfan Ye and Kai Xu and Yuhang Huang and Renjiao Yi and Zhiping Cai},
|
41 |
+
year={2024},
|
42 |
+
booktitle={AAAI}
|
43 |
+
}
|
44 |
+
```
|
45 |
+
|
46 |
+
## Files in this Repository
|
47 |
+
|
48 |
+
- `DiffusionEdge_BSDS.pt` - Model weights trained on BSDS500 dataset
|
49 |
+
|
50 |
+
## Contact and Support
|
51 |
+
|
52 |
+
For questions about the model architecture, training, or original implementation:
|
53 |
+
- **Original Authors**: Contact information available in the [original repository](https://github.com/GuHuangAI/DiffusionEdge)
|
54 |
+
|
55 |
+
For questions specific to the MooneyMaker integration:
|
56 |
+
- **MooneyMaker Project**: [Link to repo]
|
57 |
+
- [Lars C. Reining](mailto:[email protected])
|
58 |
+
|
59 |
+
## Disclaimer
|
60 |
+
|
61 |
+
- These weights are redistributed for convenience and research purposes
|
62 |
+
- All rights and ownership of the weights remain with the original authors
|
63 |
+
- Users should verify compliance with all applicable licenses and terms before use
|
64 |
+
- For the most up-to-date weights and code, always refer to the original repository
|
65 |
+
|
66 |
+
## Acknowledgments
|
67 |
+
|
68 |
+
We thank the original authors (Yunfan Ye, Yuhang Huang, Renjiao Yi, Zhiping Cai, and Kai Xu) for making their work publicly available.
|
69 |
+
|
70 |
+
---
|