updates
This commit is contained in:
parent
4320098cf5
commit
09ca721f88
2
train.py
2
train.py
|
@ -407,7 +407,7 @@ if __name__ == '__main__':
|
||||||
parser.add_argument('--bucket', type=str, default='', help='gsutil bucket')
|
parser.add_argument('--bucket', type=str, default='', help='gsutil bucket')
|
||||||
parser.add_argument('--img-weights', action='store_true', help='select training images by weight')
|
parser.add_argument('--img-weights', action='store_true', help='select training images by weight')
|
||||||
parser.add_argument('--cache-images', action='store_true', help='cache images for faster training')
|
parser.add_argument('--cache-images', action='store_true', help='cache images for faster training')
|
||||||
parser.add_argument('--weights', type=str, default='', help='initial weights') # i.e. weights/darknet.53.conv.74
|
parser.add_argument('--weights', type=str, default='weights/yolov3-spp.weights', help='initial weights') # i.e. weights/darknet.53.conv.74
|
||||||
parser.add_argument('--arc', type=str, default='default', help='yolo architecture') # defaultpw, uCE, uBCE
|
parser.add_argument('--arc', type=str, default='default', help='yolo architecture') # defaultpw, uCE, uBCE
|
||||||
parser.add_argument('--prebias', action='store_true', help='transfer-learn yolo biases prior to training')
|
parser.add_argument('--prebias', action='store_true', help='transfer-learn yolo biases prior to training')
|
||||||
parser.add_argument('--name', default='', help='renames results.txt to results_name.txt if supplied')
|
parser.add_argument('--name', default='', help='renames results.txt to results_name.txt if supplied')
|
||||||
|
|
Loading…
Reference in New Issue