Skip to content

Commit 684c503

Browse files
committed
Merge branch 'master' of github.com:google/automl
2 parents 7074952 + 891dc6c commit 684c503

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

efficientdet/README.md

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -279,6 +279,16 @@ Finetune needs to use --ckpt rather than --backbone_ckpt.
279279
--hparams=voc_config.yaml \
280280
--use_tpu=False
281281

282+
If you want to do inference for custom data, you can run
283+
284+
# Setting hparams-flag is needed sometimes.
285+
!python model_inspect.py --runmode=infer \
286+
--model_name=efficientdet-d0 --ckpt_path=efficientdet-d0 \
287+
--hparams=voc_config.yaml \
288+
--input_image=img.png --output_image_dir=/tmp/
289+
290+
You should check more details of runmode which is written in caption-4.
291+
282292
## 9. Training EfficientDets on TPUs.
283293

284294
To train this model on Cloud TPU, you will need:

0 commit comments

Comments
 (0)