From 6da62e433da39077186a0521edf3f52b9386c307 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 26 Aug 2018 20:24:37 +0200 Subject: [PATCH] updates --- test.py | 2 +- utils/datasets.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/test.py b/test.py index 014a7d29..431515ba 100644 --- a/test.py +++ b/test.py @@ -32,7 +32,7 @@ num_classes = int(data_config['classes']) model = Darknet(opt.cfg, opt.img_size) # Load weights -weights_path = 'checkpoints/yolov3.pt' +weights_path = 'checkpoints/yolov3.weights' if weights_path.endswith('.weights'): # darknet format load_weights(model, weights_path) elif weights_path.endswith('.pt'): # pytorch format diff --git a/utils/datasets.py b/utils/datasets.py index ed71c31b..07647166 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -68,7 +68,7 @@ class ListDataset(): # for training if platform == 'darwin': # macos self.img_files = [path.replace('\n', '').replace('/images', '/Users/glennjocher/Downloads/DATA/coco/images') for path in self.img_files] - else: + else: # linux (gcp cloud) self.img_files = [path.replace('\n', '').replace('/images', '../coco/images') for path in self.img_files] self.label_files = [path.replace('images', 'labels').replace('.png', '.txt').replace('.jpg', '.txt') for path in