diff --git a/data/coco_100img.data b/data/coco_100img.data index aa96ad45..716cf7c9 100644 --- a/data/coco_100img.data +++ b/data/coco_100img.data @@ -1,6 +1,6 @@ classes=80 -train=../coco/coco_100img.txt -valid=../coco/coco_100img.txt +train=./data/coco_100img.txt +valid=./data/coco_100img.txt names=data/coco.names backup=backup/ eval=coco diff --git a/data/coco_10img.data b/data/coco_10img.data index 7b291283..ea37a698 100644 --- a/data/coco_10img.data +++ b/data/coco_10img.data @@ -1,6 +1,6 @@ classes=80 -train=../coco/coco_10img.txt -valid=../coco/coco_10img.txt +train=./data/coco_10img.txt +valid=./data/coco_10img.txt names=data/coco.names backup=backup/ eval=coco diff --git a/data/coco_1cls.data b/data/coco_1cls.data index 8b390171..a19e3c0f 100644 --- a/data/coco_1cls.data +++ b/data/coco_1cls.data @@ -1,6 +1,6 @@ -classes=80 -train=../coco/coco_1cls.txt -valid=../coco/coco_1cls.txt +classes=1 +train=./data/coco_1cls.txt +valid=./data/coco_1cls.txt names=data/coco.names backup=backup/ eval=coco diff --git a/data/coco_1img.data b/data/coco_1img.data index 04142671..d97252f2 100644 --- a/data/coco_1img.data +++ b/data/coco_1img.data @@ -1,6 +1,6 @@ classes=80 -train=../coco/coco_1img.txt -valid=../coco/coco_1img.txt +train=./data/coco_1img.txt +valid=./data/coco_1img.txt names=data/coco.names backup=backup/ eval=coco diff --git a/train.py b/train.py index 7ea5b763..a62612ce 100644 --- a/train.py +++ b/train.py @@ -33,7 +33,7 @@ def train( img_size = 608 # initiate with maximum multi_scale size num_workers = 0 # bug https://github.com/ultralytics/yolov3/issues/174 else: - pass # torch.backends.cudnn.benchmark = True # unsuitable for multiscale + torch.backends.cudnn.benchmark = True # unsuitable for multiscale # Configure run train_path = parse_data_cfg(data_cfg)['train']