updates
This commit is contained in:
parent
5e79810e69
commit
2fab66607c
|
@ -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
|
||||||
|
|
|
@ -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
|
||||||
|
|
|
@ -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
|
||||||
|
|
|
@ -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
|
||||||
|
|
2
train.py
2
train.py
|
@ -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']
|
||||||
|
|
Loading…
Reference in New Issue