search

yangxue0827 / FPN_Tensorflow

A Tensorflow implementation of FPN detection framework.
416 stars 150 forks source link

This project is only detecting one class? #4

Closed duyuting closed 6 years ago

duyuting commented 6 years ago

how to detect three classess ,change the --dataset='1 2 3'?????

Kongsea commented 6 years ago

It can detect more than one class. I have used it to detect two classes successfully. But the accuracy is not as good as Facebook's Detectron.

yangxue0827 commented 6 years ago

You only need to modify the parameter CLASS_NUM in cfgs.py. However, as Kongsea say, the accuracy is not quiet well, hope you can find out the problem and contact me.

yangxue0827 commented 6 years ago

@duyuting @Kongsea https://github.com/DetectionTeamUCAS/Faster-RCNN_Tensorflow

yangxue0827 commented 5 years ago

Recommend improved code: https://github.com/DetectionTeamUCAS/FPN_Tensorflow. @duyuting