@ -183,7 +183,7 @@ def _ssd_train(args: argparse.Namespace) -> None:
|
|||||||
_init_eager_mode()
|
_init_eager_mode()
|
||||||
|
|
||||||
batch_size, image_size, learning_rate, steps_per_val_epoch, nr_classes, \
|
batch_size, image_size, learning_rate, steps_per_val_epoch, nr_classes, \
|
||||||
iou_treshold, dropout_rate, top_k, nr_trajectories, \
|
iou_threshold, dropout_rate, top_k, nr_trajectories, \
|
||||||
coco_path, summary_path, weights_path, train_gt_path, val_gt_path, \
|
coco_path, summary_path, weights_path, train_gt_path, val_gt_path, \
|
||||||
save_train_images, save_summaries = _ssd_train_get_config_values(conf.get_property)
|
save_train_images, save_summaries = _ssd_train_get_config_values(conf.get_property)
|
||||||
|
|
||||||
@ -200,7 +200,7 @@ def _ssd_train(args: argparse.Namespace) -> None:
|
|||||||
image_size,
|
image_size,
|
||||||
nr_classes,
|
nr_classes,
|
||||||
"training",
|
"training",
|
||||||
iou_treshold,
|
iou_threshold,
|
||||||
dropout_rate,
|
dropout_rate,
|
||||||
top_k,
|
top_k,
|
||||||
pre_trained_weights_file)
|
pre_trained_weights_file)
|
||||||
|
|||||||
Reference in New Issue
Block a user