Update README.md
Browse files
README.md
CHANGED
|
@@ -1,5 +1,8 @@
|
|
| 1 |
---
|
| 2 |
license: mit
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
| 4 |
# GeoDiffusion Model Card
|
| 5 |
GeoDiffusion is a detection data generation model fine-tuned from [Stable Diffusion v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5). This repo contains the GeoDiffusion checkpoint trained on **nuImages** dataset under **512x512** with conditional generation support for time-of-day and weather. Check detailed usage in our [Github repo](https://github.com/KaiChen1998/GeoDiffusion).
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
+
datasets:
|
| 4 |
+
- KaiChen1998/nuimages-geodiffusion
|
| 5 |
+
library_name: diffusers
|
| 6 |
---
|
| 7 |
# GeoDiffusion Model Card
|
| 8 |
GeoDiffusion is a detection data generation model fine-tuned from [Stable Diffusion v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5). This repo contains the GeoDiffusion checkpoint trained on **nuImages** dataset under **512x512** with conditional generation support for time-of-day and weather. Check detailed usage in our [Github repo](https://github.com/KaiChen1998/GeoDiffusion).
|