|
|
@@ -48,13 +48,14 @@ def parse_args(): |
|
|
parser.add_argument('--local_rank', type=int, default=0, help='current rank to support distributed') |
|
|
parser.add_argument('--local_rank', type=int, default=0, help='current rank to support distributed') |
|
|
parser.add_argument('--world_size', type=int, default=1, help='current process number to support distributed') |
|
|
parser.add_argument('--world_size', type=int, default=1, help='current process number to support distributed') |
|
|
|
|
|
|
|
|
args, _ = parser.parse_known_args() |
|
|
|
|
|
|
|
|
arg, _ = parser.parse_known_args() |
|
|
|
|
|
|
|
|
return args |
|
|
|
|
|
|
|
|
return arg |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def val(args): |
|
|
|
|
|
'''eval''' |
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
|
|
|
args = parse_args() |
|
|
|
|
|
|
|
|
print('=============yolov3 start evaluating==================') |
|
|
print('=============yolov3 start evaluating==================') |
|
|
|
|
|
|
|
|
# logger |
|
|
# logger |
|
|
@@ -190,7 +191,7 @@ def val(args): |
|
|
|
|
|
|
|
|
aps_str = '' |
|
|
aps_str = '' |
|
|
for cls in evaluate: |
|
|
for cls in evaluate: |
|
|
per_line, = plt.plot(evaluate[cls]['recall'], evaluate[cls]['presicion'], 'b-') |
|
|
|
|
|
|
|
|
per_line, = plt.plot(evaluate[cls]['recall'], evaluate[cls]['precision'], 'b-') |
|
|
per_line.set_label('%s:AP=%.3f' % (cls, evaluate[cls]['ap'])) |
|
|
per_line.set_label('%s:AP=%.3f' % (cls, evaluate[cls]['ap'])) |
|
|
aps_str += '_%s_AP_%.3f' % (cls, evaluate[cls]['ap']) |
|
|
aps_str += '_%s_AP_%.3f' % (cls, evaluate[cls]['ap']) |
|
|
plt.plot([i / 1000.0 for i in range(1, 1001)], [i / 1000.0 for i in range(1, 1001)], 'y--') |
|
|
plt.plot([i / 1000.0 for i in range(1, 1001)], [i / 1000.0 for i in range(1, 1001)], 'y--') |
|
|
@@ -208,8 +209,3 @@ def val(args): |
|
|
plt.savefig(ap_save_path) |
|
|
plt.savefig(ap_save_path) |
|
|
|
|
|
|
|
|
print('=============yolov3 evaluating finished==================') |
|
|
print('=============yolov3 evaluating finished==================') |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
|
|
|
arg = parse_args() |
|
|
|
|
|
val(arg) |
|
|
|