search

lsrock1 / maskrcnn_benchmark.cpp

Implementation maskrcnn-benchmark, pytorch c++ frontend
MIT License
81 stars 23 forks source link

anchors #12

Closed Rguoo closed 5 years ago

Rguoo commented 5 years ago

hi，how can i generate anchors with libtorch on gpu device.

lsrock1 commented 5 years ago

Anchor is generated in https://github.com/lsrock1/maskrcnn_benchmark.cpp/blob/master/source/rcnn/modeling/rpn/anchor_generator.cpp

Rguoo commented 5 years ago

@lsrock1 Thankes！