Simplify label map - Githubissues

DIUx-xView / xview3-reference

Reference data processing code and model for the xView3 prize challenge.

Other

44 stars 27 forks source link

Closed jdunnmon closed 3 years ago

jdunnmon commented 3 years ago

Simplifies label map in dataloader.py to reduce confusion (no functional changes)
Adjusts baseline model num_classes kwarg calls to use reference label map values rather than keys -- no effect on previously trained models, but eliminates unused classes in model output
Updates inference.py, train_reference.ipynb, and train.py to reflect these changes

jdunnmon commented 3 years ago

Should also address #2

RitwikGupta commented 3 years ago

Verified on train, inference, and metric scripts, as well as reference notebook.