File size: 2,039 Bytes
4457836 a229cde fc3d3e7 4457836 a229cde fc3d3e7 a229cde fc3d3e7 a229cde fc3d3e7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
---
license: mit
tags:
- Astronomy
- Classification
- Object Detection
---
<div align="center">
<h1 style="font-size:40px;font-weight:bold">DRAFTS</h1>
_✨ Deep learning-based RAdio Fast Transient Search pipeline✨_
<img src="https://counter.seku.su/cmoe?name=APOD&theme=r34" /><br>
</div>
## <div align="center">Description</div>
Here is the model repository for the Deep learning-based RAdio Fast Transient Search pipeline ([DRAFTS](https://github.com/SukiYume/DRAFTS)).
We invite you to stay tuned for updates on the remaining components and different versions of the models.
## <div align="center">Usage</div>
There are four `.pth` files in this repository.
### Object Detection
The files starting with `cent` are the trained model checkpoints for the object detection models in the [DRAFTS](https://github.com/SukiYume/DRAFTS) project.
To load the object detection model
```python
import torch
from centernet_model import centernet
base_model = 'resnet18' # 'resnet50'
model = centernet(model_name=base_model)
model.load_state_dict(torch.load('cent_{}.pth'.format(base_model)))
model.eval()
```
### Binary Classification
The files starting with `class` are the trained model checkpoints for the classification models in the [DRAFTS](https://github.com/SukiYume/DRAFTS) project.
To load the classification detection model
```python
import torch
from binary_model import BinaryNet
base_model = 'resnet18' # 'resnet50'
model = BinaryNet(base_model, num_classes=2)
model.load_state_dict(torch.load('class_{}.pth'.format(base_model)))
model.eval()
```
## <div align="center">Contributing</div>
We welcome contributions to the DRAFTS project! If you have any suggestions, bug reports, or feature requests, please open an issue or submit a pull request on our [GitHub repository](https://github.com/SukiYume/DRAFTS).
## <div align="center">Contact</div>
For any questions or inquiries, please contact us at [email protected] or [email protected]
<div align="center">
✨ Thank you for using DRAFTS! ✨
</div>
|