Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-05-18 23:24:26 +02:00
parent aae93a9651
commit b034382c8b
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -138,7 +138,7 @@ def train(
# plt.savefig('LR.png', dpi=300)
# Dataset
dataset = LoadImagesAndLabels(train_path, img_size, batch_size, augment=False, rect=False, image_weights=False)
dataset = LoadImagesAndLabels(train_path, img_size, batch_size, augment=True, rect=False, image_weights=False)
# Initialize distributed training
if torch.cuda.device_count() > 1:

6
utils/gcp.sh
View File

@ -61,8 +61,10 @@ sudo shutdown
# Unit tests
python3 detect.py # detect 2 persons, 1 tie
python3 test.py --data data/coco_32img.data # test mAP = 0.78
python3 train.py --data data/coco_32img.data --epochs 4 --nosave # train 4 epochs
python3 test.py --data data/coco_32img.data # test mAP = 0.8
python3 train.py --data data/coco_32img.data --epochs 5 --nosave # train 5 epochs
python3 train.py --data data/coco_1cls.data --epochs 5 --nosave # train 5 epochs
python3 train.py --data data/coco_1img.data --epochs 5 --nosave # train 5 epochs
# AlexyAB Darknet
./darknet detector train ../supermarket2/supermarket2.data cfg/yolov3-spp-sm2.cfg darknet53.conv.74 # train