microsoft / torchgeo

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data
https://www.osgeo.org/projects/torchgeo/
MIT License
2.64k stars 326 forks source link

Object Detection: Add support for pretrained `weights` #1044

Open adamjstewart opened 1 year ago

adamjstewart commented 1 year ago

Summary

We should add support for using pretrained backbones to our semantic segmentation and object detection trainers.

Rationale

Our classification/regression/BYOL trainers support pretrained weights, we should be uniform.

Implementation

Need to figure out if any of these models support the use of timm for backbones or whether or not we can manually replace the backbone.

Alternatives

We currently support ImageNet pre-trained backbones for a limited subset, but that isn't the same.

Additional information

No response

adamjstewart commented 1 year ago

Update: segmentation weights are now supported but not detection weights.