sovit-123 / fasterrcnn-pytorch-training-pipeline

PyTorch Faster R-CNN Object Detection on Custom Dataset
MIT License
223 stars 77 forks source link

Integrate PyGAD genetic algorithm #130

Open Zhihao1002 opened 9 months ago

Zhihao1002 commented 9 months ago

Hi just want to explore the possibility of integrating the PyGAD GA to optimise the model. If you have any chance working on PyGAD, would you mind to guide on how to insert into the current code. Thanks

sovit-123 commented 9 months ago

Hi. I have never used PyGAD before. Please shed some light here. Is there something we can do with object detection + PyGAD?

Zhihao1002 commented 9 months ago

the initial idea was to use GA to identify the highest fitness value of parameter arrangement to improve the accuracy of the model, but not really sure how it can be done.

sovit-123 commented 9 months ago

I will try to look imto it.