update drafts model

README.md

@@ -3,20 +3,69 @@ 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 (Under Construction)✨_
+_✨ 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)).
 
-Currently, we have uploaded the TensorFlow version of the binary classification model for the DRAFTS pipeline. 
 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 ykzhang@nao.cas.cn or ykzhang@escape.ac.cn
+
+<div align="center">
+✨ Thank you for using DRAFTS! ✨
+</div>

BinaryModel-TFNVer.h5 → cent_resnet18.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:67276295fb3a636ed57cfbb75cef4dbe710ffed3c54829fcae3b5ea16c2f19f7
-size 289753440
+oid sha256:85c97ce654222410ff1f581e80302ebddfe3479f65bcb03a867068c9806d650f
+size 75519014

cent_resnet50.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:510f5b6b3049567178d4aa7c22ec9d37d378f165ee540b02b82e1cc58a6b100d
+size 153397676

class_resnet18.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2c9c6b4b990596bd07ddfb5527c21cd689247d5026b2815bd1b6ce5294e66cc6
+size 44766392

class_resnet50.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5b75c14256fc1909dca7ecea2f4e2c3af05eff400d4d615c5249b90e39325acc
+size 94347198