tianzhi0549 / FCOS

FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)
https://arxiv.org/abs/1904.01355
Other
3.28k stars 630 forks source link

is there any tensorflow or caffe project for FCOS #103

Open YoungerGao opened 5 years ago

YoungerGao commented 5 years ago

thanks for your work, very interesting mind. but for the online service, the c++ api interface is needed, is there any tensorflow or caffe project for FCOS?

tianzhi0549 commented 5 years ago

@YoungerGao There are some reimplementations based on TensorFlow. Just google them.

YoungerGao commented 5 years ago

@tianzhi0549 well, I will try to solve the tensorflow project, thanks

xuannianz commented 5 years ago

@YoungerGao I implemented FCOS in keras and tensorflow based on this official project. https://github.com/xuannianz/keras-fcos

YoungerGao commented 5 years ago

@YoungerGao I implemented FCOS in keras and tensorflow based on this official project. https://github.com/xuannianz/keras-fcos

Great job!!! Have you check the difference of mAP between your tensorflow project and the official project?

xuannianz commented 5 years ago

@YoungerGao No, I only test on Pascal VOC. I know there's room for improvement, so I open sourced it and would like to get help from guys who are interested in it.

YoungerGao commented 5 years ago

@YoungerGao No, I only test on Pascal VOC. I know there's room for improvement, so I open sourced it and would like to get help from guys who are interested in it.

Got it, thanks for your reply

MsRabbits commented 5 years ago

@YoungerGao I implemented FCOS in keras and tensorflow based on this official project. https://github.com/xuannianz/keras-fcos

Thx for your project about focs with tensorflow. But there seems weird when i train my own dataset with your code: after 2 epochs, the regression_loss become nan, could you give me some advice about that?