From 8583072cc69cff21e546c6171dd0fdf47239f0c1 Mon Sep 17 00:00:00 2001 From: Jarrett Yeo <32927034+jarrettyeo@users.noreply.github.com> Date: Tue, 18 Feb 2020 02:03:38 +0800 Subject: [PATCH] Update object_names_array argument for trainer.setTrainConfig --- imageai/Detection/Custom/CUSTOMDETECTIONTRAINING.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/imageai/Detection/Custom/CUSTOMDETECTIONTRAINING.md b/imageai/Detection/Custom/CUSTOMDETECTIONTRAINING.md index dc023ddc..4b499825 100644 --- a/imageai/Detection/Custom/CUSTOMDETECTIONTRAINING.md +++ b/imageai/Detection/Custom/CUSTOMDETECTIONTRAINING.md @@ -94,7 +94,7 @@ trainer.setTrainConfig(object_names_array=["hololens"], batch_size=4, num_experi In the line above, we configured our detection model trainer. The parameters we stated in the function as as below: -- **num_objects** : this is an array containing the names of the objects in our dataset +- **object_names_array** : this is an array containing the names of the objects in our dataset - **batch_size** : this is to state the batch size for the training - **num_experiments** : this is to state the number of times the network will train over all the training images, which is also called epochs