Visual-Behavior / aloception-oss

Aloception is a set of package for computer vision: aloscene, alodataset, alonet.
Other
92 stars 7 forks source link

load model from run_id #281

Closed anhtu293 closed 1 year ago

anhtu293 commented 1 year ago

General description of your pull request with the list of new features and/or bugs.

model = DetrR50(num_classes=91, weights="detr-r50")

load from downloaded weight

load_weights(model, weights="detr-r50")

load from .pth file

load_weights(model, weights="~/.aloception/weights/detr-r50/detr-r50.pth")

load from run_id

load_weights(model, run_id="your_run_id_heer", project_run_id="your_project_run_id",)



_____
This pull request includes

- [ ] Bug fix (non-breaking change which fixes an issue)
- [x] New feature (non-breaking change which adds functionality)
- [ ] Breaking change (fix or feature that would cause existing functionality to not work as expected)
- [ ] This change requires a documentation update