search

Pirazh / Vehicle_Key_Point_Orientation_Estimation

This is the repository for Vehicle Key-Point and Orientation Estimation
81 stars 18 forks source link

about the code #9

Closed xieyuhaoli closed 3 years ago

xieyuhaoli commented 3 years ago

in the utilities.py file,in your test ,do you use the code : if target[n, c, 0] > 1 and target[n, c, 1] > 1:

Pirazh commented 3 years ago

Yes, during the testing stage, the code only considers the key-points which their ground-truth coordinates exist.

xieyuhaoli commented 3 years ago

Yes, during the testing stage, the code only considers the key-points which their ground-truth coordinates exist. In the second stage, the direction estimation has no effect on the detection of key points. In the second stage, the accuracy of the detection of key points is modified only through a section of network, right?

Pirazh commented 3 years ago

The refinement stage tries to yield cleaner heat-maps that outline the location of key-points. To train the refinement stage, the weights of the course stage estimator are frozen. In this stage, refinement of key-points and orientation estimation is done simultaneously.