This commit is contained in:
Glenn Jocher 2019-06-21 23:11:24 +02:00
parent 4f7fee45ff
commit 1a0385c77d
2 changed files with 4 additions and 2 deletions

View File

@ -269,7 +269,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing
img, ratiow, ratioh, padw, padh = letterbox(img, new_shape=shape, mode='rect')
else:
shape = self.img_size
img, ratiow, ratioh, padw, padh = letterbox(img, new_shape=shape, mode='scaleFill')
img, ratiow, ratioh, padw, padh = letterbox(img, new_shape=shape, mode='square')
# Load labels
labels = []

View File

@ -95,6 +95,8 @@ python3 test.py --data ../supermarket2/supermarket2.data --weights ../darknet/ba
# Debug/Development
python3 train.py --data data/coco.data --img-size 416 --batch-size 8 --accumulate 8
python3 train.py --data data/coco.data --img-size 320 --single-scale --batch-size 32 --accumulate 2 --epochs 1
python3 test.py --weights weights/latest.pt --cfg cfg/yolov3-spp.cfg --img-size 320
gsutil cp evolve.txt gs://ultralytics
sudo shutdown