@@ -74,7 +74,7 @@ View `config.py` for more argument details (batch size, learning rate, etc). | |||
* Evaluate parking-slot detection | |||
```(shell) | |||
python evaluate.py --label_directory $LABEL_DIRECTORY --image_directory $IMAGE_DIRECTORY --detector_weights $DETECTOR_WEIGHTS | |||
python ps_evaluate.py --label_directory $LABEL_DIRECTORY --image_directory $IMAGE_DIRECTORY --detector_weights $DETECTOR_WEIGHTS | |||
``` | |||
`LABEL_DIRECTORY` is the directory containing testing json labels. |
@@ -33,6 +33,7 @@ SLOT_SUPPRESSION_DOT_PRODUCT_THRESH = 0.8 | |||
# precision = 0.995585, recall = 0.995805 | |||
CONFID_THRESH_FOR_POINT = 0.11676871 | |||
def add_common_arguments(parser): | |||
"""Add common arguments for training and inference.""" | |||
parser.add_argument('--detector_weights', |
@@ -34,7 +34,8 @@ def collect_error(ground_truths, predictions, thresh): | |||
if idx >= 0: | |||
detected_point = predictions[idx][1] | |||
dists.append(calc_point_squre_dist(detected_point, ground_truth)) | |||
angles.append(calc_point_direction_angle(detected_point, ground_truth)) | |||
angles.append(calc_point_direction_angle( | |||
detected_point, ground_truth)) | |||
else: | |||
continue | |||
return dists, angles |