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

updates

This commit is contained in:
Glenn Jocher 2018-12-10 13:19:13 +01:00
parent 362b41436a
commit 3fe3951268
3 changed files with 22 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
detect.py
View File

@ -21,7 +21,6 @@ def detect(
save_txt=False, save_txt=False,
save_images=False, save_images=False,
): ):
device = torch_utils.select_device() device = torch_utils.select_device()
print("Using device: \"{}\"".format(device)) print("Using device: \"{}\"".format(device))
@ -150,7 +149,6 @@ if __name__ == '__main__':
parser.add_argument('--output-folder', type=str, default='output', help='path to outputs') parser.add_argument('--output-folder', type=str, default='output', help='path to outputs')
parser.add_argument('--plot-flag', type=bool, default=True) parser.add_argument('--plot-flag', type=bool, default=True)
parser.add_argument('--txt-out', type=bool, default=False) parser.add_argument('--txt-out', type=bool, default=False)
parser.add_argument('--cfg', type=str, default='cfg/yolov3.cfg', help='cfg file path') parser.add_argument('--cfg', type=str, default='cfg/yolov3.cfg', help='cfg file path')
parser.add_argument('--data-config', type=str, default='cfg/coco.data', help='path to data config file') parser.add_argument('--data-config', type=str, default='cfg/coco.data', help='path to data config file')
parser.add_argument('--conf-thres', type=float, default=0.50, help='object confidence threshold') parser.add_argument('--conf-thres', type=float, default=0.50, help='object confidence threshold')

19
test.py
View File

@ -8,15 +8,15 @@ from utils import torch_utils
def test( def test(
net_config_path, net_config_path,
data_config_path, data_config_path,
weights_file_path, weights_file_path,
batch_size=16, batch_size=16,
img_size=416, img_size=416,
iou_thres=0.5, iou_thres=0.5,
conf_thres=0.3, conf_thres=0.3,
nms_thres=0.45, nms_thres=0.45,
n_cpus=0, n_cpus=0,
): ):
device = torch_utils.select_device() device = torch_utils.select_device()
print("Using device: \"{}\"".format(device)) print("Using device: \"{}\"".format(device))
@ -126,7 +126,6 @@ def test(
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser(prog='test.py') parser = argparse.ArgumentParser(prog='test.py')
parser.add_argument('--batch-size', type=int, default=32, help='size of each image batch') parser.add_argument('--batch-size', type=int, default=32, help='size of each image batch')
parser.add_argument('--cfg', type=str, default='cfg/yolov3.cfg', help='path to model config file') parser.add_argument('--cfg', type=str, default='cfg/yolov3.cfg', help='path to model config file')

30
train.py
View File

@ -12,25 +12,22 @@ from utils import torch_utils
import test import test
DARKNET_WEIGHTS_FILENAME = 'darknet53.conv.74' DARKNET_WEIGHTS_FILENAME = 'darknet53.conv.74'
DARKNET_WEIGHTS_URL = 'https://pjreddie.com/media/files/{}'.format( DARKNET_WEIGHTS_URL = 'https://pjreddie.com/media/files/{}'.format(DARKNET_WEIGHTS_FILENAME)
DARKNET_WEIGHTS_FILENAME
)
def train( def train(
net_config_path, net_config_path,
data_config_path, data_config_path,
img_size=416, img_size=416,
resume=False, resume=False,
epochs=100, epochs=100,
batch_size=16, batch_size=16,
weights_path='weights', weights_path='weights',
report=False, report=False,
multi_scale=False, multi_scale=False,
freeze_backbone=True, freeze_backbone=True,
var=0, var=0,
): ):
device = torch_utils.select_device() device = torch_utils.select_device()
print("Using device: \"{}\"".format(device)) print("Using device: \"{}\"".format(device))
@ -236,13 +233,12 @@ def train(
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--epochs', type=int, default=100, help='number of epochs') parser.add_argument('--epochs', type=int, default=100, help='number of epochs')
parser.add_argument('--batch-size', type=int, default=16, help='size of each image batch') parser.add_argument('--batch-size', type=int, default=16, help='size of each image batch')
parser.add_argument('--data-config', type=str, default='cfg/coco.data', help='path to data config file') parser.add_argument('--data-config', type=str, default='cfg/coco.data', help='path to data config file')
parser.add_argument('--cfg', type=str, default='cfg/yolov3.cfg', help='cfg file path') parser.add_argument('--cfg', type=str, default='cfg/yolov3.cfg', help='cfg file path')
parser.add_argument('--multi-scale', default=False, help='random image sizes per batch 320 - 608') parser.add_argument('--multi-scale', action='store_true', help='random image sizes per batch 320 - 608')
parser.add_argument('--img-size', type=int, default=32 * 13, help='pixels') parser.add_argument('--img-size', type=int, default=32 * 13, help='pixels')
parser.add_argument('--weights-path', type=str, default='weights', help='path to store weights') parser.add_argument('--weights-path', type=str, default='weights', help='path to store weights')
parser.add_argument('--resume', action='store_true', help='resume training flag') parser.add_argument('--resume', action='store_true', help='resume training flag')