This commit is contained in:
Glenn Jocher 2019-04-05 16:08:18 +02:00
parent 5e79810e69
commit 2fab66607c
5 changed files with 10 additions and 10 deletions

View File

@ -1,6 +1,6 @@
classes=80 classes=80
train=../coco/coco_100img.txt train=./data/coco_100img.txt
valid=../coco/coco_100img.txt valid=./data/coco_100img.txt
names=data/coco.names names=data/coco.names
backup=backup/ backup=backup/
eval=coco eval=coco

View File

@ -1,6 +1,6 @@
classes=80 classes=80
train=../coco/coco_10img.txt train=./data/coco_10img.txt
valid=../coco/coco_10img.txt valid=./data/coco_10img.txt
names=data/coco.names names=data/coco.names
backup=backup/ backup=backup/
eval=coco eval=coco

View File

@ -1,6 +1,6 @@
classes=80 classes=1
train=../coco/coco_1cls.txt train=./data/coco_1cls.txt
valid=../coco/coco_1cls.txt valid=./data/coco_1cls.txt
names=data/coco.names names=data/coco.names
backup=backup/ backup=backup/
eval=coco eval=coco

View File

@ -1,6 +1,6 @@
classes=80 classes=80
train=../coco/coco_1img.txt train=./data/coco_1img.txt
valid=../coco/coco_1img.txt valid=./data/coco_1img.txt
names=data/coco.names names=data/coco.names
backup=backup/ backup=backup/
eval=coco eval=coco

View File

@ -33,7 +33,7 @@ def train(
img_size = 608 # initiate with maximum multi_scale size img_size = 608 # initiate with maximum multi_scale size
num_workers = 0 # bug https://github.com/ultralytics/yolov3/issues/174 num_workers = 0 # bug https://github.com/ultralytics/yolov3/issues/174
else: else:
pass # torch.backends.cudnn.benchmark = True # unsuitable for multiscale torch.backends.cudnn.benchmark = True # unsuitable for multiscale
# Configure run # Configure run
train_path = parse_data_cfg(data_cfg)['train'] train_path = parse_data_cfg(data_cfg)['train']