[WIP] Add Polygon model support #837
Conversation
|
Thanks for the contribution on this @Om-Doiphode! The code base has moved on in the past year and we've been doing some of this work elsewhere, so I'm going to go ahead and close this. @Om-Doiphode if you still want to work on this let me know and I can link you up with the folks currently working in this space |
|
Yes, I would like to work on this. |
|
Thanks @Om-Doiphode @jveitchmichaelis - @Om-Doiphode work on polygon support for us in Fall 2024 and then we lost track of it. Can you catch him up on where things are and see if he can be of help in the work you've been doing to generalize everything to points and polygons? |
|
Hi @Om-Doiphode, First thing would be to align the polygon dataloader with #1311 (ie inherit from the base dataset, modify only the bits that need adjusting). We would also need to check what settings the augmentation pipeline needs. And then add unit tests for that. I think we probably also need to look into our utilities for to/from COCO format. The evaluation code should already work with polygons, but we don't have any testing in place yet. Once these two are in place, we can look at model training. It's easier that we have the eval stage working first, because we can sanity check using any dataset (e.g. MS-COCO). I will think about a good training dataset, but probably something like OAM-TCD as I'm familiar. |
|
Since this branch is way behind the main branch (338 commits behind), should I go with |
|
I would either rebase or start from scratch and copy your changes in. |
|
I have issued a new PR here: #1359 |
|
this is superceded by 1359 |
Fixes issue #758
deepforest/models/directory.