search

microsoft / torchgeo

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

Update num_classes parameter description in ObjectDetectionTask class #2101

Closed robmarkcole closed 4 weeks ago

robmarkcole commented 4 weeks ago

adds num_classes. - Number of prediction classes (including the background) to the ObjectDetectionTask docs,

robmarkcole commented 4 weeks ago

Considering in_channels does nothing currently, is it worth mentioning? Ref https://github.com/microsoft/torchgeo/issues/1156

robmarkcole commented 4 weeks ago

@adamjstewart done