Fix val.py study plot (#4831)

* Fix val.py study plot

* call plot_val_study

* Rename plot_study_txt to plot_val_study
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-09-16 17:55:58 +02:00
1 parent 3beb871ba4
commit 27a4736e96
2 files changed
+9 -6

No files matched your search

+2 -2
View File
@@ -26,7 +26,7 @@ from utils.general import coco80_to_coco91_class, check_dataset, check_img_size,
check_suffix, check_yaml, box_iou, non_max_suppression, scale_coords, xyxy2xywh, xywh2xyxy, set_logging, \
increment_path, colorstr
from utils.metrics import ap_per_class, ConfusionMatrix
from utils.plots import plot_images, output_to_target, plot_study_txt
from utils.plots import output_to_target, plot_images, plot_val_study
from utils.torch_utils import select_device, time_sync
from utils.callbacks import Callbacks
@@ -348,7 +348,7 @@ def main(opt):
y.append(r + t) # results and times
np.savetxt(f, y, fmt='%10.4g') # save
os.system('zip -r study.zip study_*.txt')
plot_study_txt(x=x) # plot
plot_val_study(x=x) # plot
if __name__ == "__main__":