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

train.py freeze-darknet53 shortened to freeze and action store_true 

Traing with freeze: python train.py --freeze
Train without freeze: python train.py

Note: in the actual version freeze is only for first epoche
This commit is contained in:
Guillermo García 2018-12-05 15:55:09 +01:00
parent 868a116750
commit d03ce45da5
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@ -247,7 +247,7 @@ if __name__ == '__main__':
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('--report', action='store_true', help='report TP, FP, FN, P and R per batch (slower)')
parser.add_argument('--freeze-darknet53', default=False, help='freeze darknet53.conv.74 layers for first epoch')
parser.add_argument('--freeze', action='store_true', help='freeze darknet53.conv.74 layers for first epoche')
parser.add_argument('--var', type=float, default=0, help='optional test variable')
opt = parser.parse_args()
print(opt, end='\n\n')
@ -265,6 +265,6 @@ if __name__ == '__main__':
weights_path=opt.weights_path,
report=opt.report,
multi_scale=opt.multi_scale,
freeze_backbone=opt.freeze_darknet53,
freeze_backbone=opt.freeze,
var=opt.var,
)