search

johnsk95 / PT4AL

Official PyTorch implementation of "PT4AL: Using Self-Supervised Pretext Tasks for Active Learning (ECCV2022)"
54 stars 6 forks source link

Extending current work #2

Closed abhigoku10 closed 2 years ago

abhigoku10 commented 2 years ago

@johnsk95 thanks for sharing the code base, i have a query can we use the existing source code to extend it to task like object detection and segmentation? if so what changes have to be made Thanks in advance

johnsk95 commented 2 years ago

All you have to do is change the main task part (main.py) to the specific task you want to use. (e.g. segmentation)