wdoppenberg / crater-detection

Autonomous Lunar Orbit Navigation Using Ellipse R-CNN and Crater Pattern Matching
https://repository.tudelft.nl/islandora/object/uuid%3A75d27259-bcde-4caf-b3b1-01ff8d41b59f
MIT License
26 stars 6 forks source link

Replace NumPy with PyTorch for all subsystems #8

Open wdoppenberg opened 3 years ago

wdoppenberg commented 3 years ago

For convenience, acceleration, and consistency, it is favourable to have all tensor operations written in PyTorch.

It is somewhat limited by PyTorch's ability to handle broadcasted linear algebra compared to NumPy, but release 1.9 promises a better linalg module.

Liuk2072 commented 2 years ago

May I ask how to download SurRender Software, it seems to be not open source

Liuk2072 commented 2 years ago

Or do you have an open source ellipse dataset? If there is, can you please send me a copy, thank you very much.

wdoppenberg commented 2 years ago

Or do you have an open source ellipse dataset? If there is, can you please send me a copy, thank you very much.

If you want to do general ellipse detection, I'd recommend you look at https://github.com/wdoppenberg/ellipse-rcnn . It's not perfect yet, but it's a more general implementation of Ellipse-RCNN.

For an open source ellipse detection dataset, look here: http://vis-www.cs.umass.edu/fddb/

Liuk2072 commented 2 years ago

Hello, I generate .ckpt file after training, how can I use it to verify my training results?