From 51a83d36dbfc8d1f6fa546766f00f1fde60e285e Mon Sep 17 00:00:00 2001 From: ymir Date: Wed, 18 May 2022 18:40:30 +0800 Subject: [PATCH 1/3] bug fix --- det-yolov4-training/train_watcher.py | 1 + 1 file changed, 1 insertion(+) diff --git a/det-yolov4-training/train_watcher.py b/det-yolov4-training/train_watcher.py index eb20861..3d47a6d 100644 --- a/det-yolov4-training/train_watcher.py +++ b/det-yolov4-training/train_watcher.py @@ -75,6 +75,7 @@ def _on_train_log_yaml_modified(self, src_path: str) -> None: if isinstance(mean_ap, float): self._tensorboard_writer.add_scalar(tag="train/mAP", scalar_value=mean_ap, global_step=iteration) if class_aps and isinstance(class_aps, dict): + class_aps = {str(k): v for k, v in class_aps.items()} self._tensorboard_writer.add_scalars(main_tag='train/aps', tag_scalar_dict=class_aps, global_step=iteration) self._tensorboard_writer.flush() From aa9514280da7c4688e2ae112c46ec7b8e0738e6f Mon Sep 17 00:00:00 2001 From: ymir Date: Wed, 18 May 2022 21:09:53 +0800 Subject: [PATCH 2/3] update --- det-yolov4-training/train_watcher.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/det-yolov4-training/train_watcher.py b/det-yolov4-training/train_watcher.py index 3d47a6d..4156e21 100644 --- a/det-yolov4-training/train_watcher.py +++ b/det-yolov4-training/train_watcher.py @@ -1,7 +1,7 @@ import logging import os import re -from typing import Callable, Tuple +from typing import Callable, List, Tuple from tensorboardX import SummaryWriter import yaml @@ -21,7 +21,7 @@ def __init__(self, width: int, height: int, class_num: int) -> None: self._class_numbers = class_num self._tensorboard_writer = SummaryWriter(log_dir='/out/tensorboard') - self._pattern_and_handlers: Tuple[str, Callable] = [ + self._pattern_and_handlers: List[Tuple[str, Callable]] = [ ('^.*best.weights$', _DarknetTrainingHandler._on_best_weights_modified), ('^.*train-log.yaml$', _DarknetTrainingHandler._on_train_log_yaml_modified) ] From 6c2bb4f57f1f90acd6886c2495a8bd51c1ac1051 Mon Sep 17 00:00:00 2001 From: ymir Date: Thu, 19 May 2022 10:16:35 +0800 Subject: [PATCH 3/3] update --- det-yolov4-training/src/detector.c | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/det-yolov4-training/src/detector.c b/det-yolov4-training/src/detector.c index 6d5648a..d23d582 100644 --- a/det-yolov4-training/src/detector.c +++ b/det-yolov4-training/src/detector.c @@ -389,7 +389,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i if (names_size > 0) { fprintf(file_ptr, "aps:\n"); for (int idx = 0; idx < names_size; ++idx) { - fprintf(file_ptr, " %s: %f\n", names[idx], aps[idx]); + fprintf(file_ptr, " \'%s\': %f\n", names[idx], aps[idx]); } } fclose(file_ptr);