diff --git a/detect.py b/detect.py index 827389bd..4e4f466d 100755 --- a/detect.py +++ b/detect.py @@ -145,3 +145,5 @@ def main(opt): if __name__ == '__main__': torch.cuda.empty_cache() main(opt) + if platform == 'darwin': # MacOS (local) + os.system('open ' + opt.output_folder) diff --git a/utils/gcp.sh b/utils/gcp.sh index 691e1257..88c88eaa 100644 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -1,7 +1,7 @@ #!/usr/bin/env bash # Start -sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3 && python3 train.py +sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3 && python3 train.py -epochs 2 # Resume python3 train.py -resume 1 diff --git a/utils/utils.py b/utils/utils.py index f62eca34..0a4c6f02 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -436,7 +436,7 @@ def plot_results(): import matplotlib.pyplot as plt plt.figure(figsize=(16, 8)) s = ['X', 'Y', 'Width', 'Height', 'Objectness', 'Classification', 'Total Loss', 'Precision', 'Recall', 'mAP'] - for f in ('results.txt', + for f in ('results5.txt','results_new.txt','results3.txt', ): results = np.loadtxt(f, usecols=[2, 3, 4, 5, 6, 7, 8, 9, 10]).T # column 16 is mAP for i in range(9):