diff --git a/utils/datasets.py b/utils/datasets.py index e28c6da5..66c9feb7 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -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 = [] diff --git a/utils/gcp.sh b/utils/gcp.sh index 34d7c1c0..9780baa4 100755 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -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