jamycheung / Trans4PASS

Repository of Trans4PASS (accepted to CVPR2022)
Apache License 2.0
75 stars 17 forks source link

simplified version #9

Closed W-Kinlon closed 1 year ago

W-Kinlon commented 1 year ago

Hi, I want to use Trans4PASS to train my own dataset and obtain my own set of metrics. Could you provide a simplified version of the network that only requires input of size and segmentation type?

jamycheung commented 1 year ago

Hi, thanks for your interest.

If you want to train on your onw dataset, one possible solution is to modify one of the dataloaders (check here). About the metric, it would be helpful to check existing metrics.

If I understand correctly, you could create demo scripts regarding eval_xx.py files here.