Closed6

EfficientDet-lite0,lite1,lite2,lite3,lite4

$ python3 keras/inspector.py \
  --mode=export \
  --model_name=efficientdet-lite0 \
  --saved_model_dir=saved_model_lite0 \
  --tflite=FP32 \
  --image_size=320
$ python3 keras/inspector.py \
  --mode=export \
  --model_name=efficientdet-lite1 \
  --saved_model_dir=saved_model_lite1 \
  --tflite=FP32 \
  --image_size=384
$ python3 keras/inspector.py \
  --mode=export \
  --model_name=efficientdet-lite2 \
  --saved_model_dir=saved_model_lite2 \
  --tflite=FP32 \
  --image_size=448
$ python3 keras/inspector.py \
  --mode=export \
  --model_name=efficientdet-lite3 \
  --saved_model_dir=saved_model_lite3 \
  --tflite=FP32 \
  --image_size=512
$ python3 keras/inspector.py \
  --mode=export \
  --model_name=efficientdet-lite4 \
  --saved_model_dir=saved_model_lite4 \
  --tflite=FP32 \
  --image_size=512
python3 main.py \
    --mode=train_and_eval \
    --train_file_pattern="tfrecord/pascal-00000-of-00100.tfrecord"  \
    --val_file_pattern="tfrecord/pascal-00000-of-00100.tfrecord"  \
    --model_name=efficientdet-lite0 \
    --model_dir=efficientdet-lite0  \
    --backbone_ckpt=efficientnet-lite0 \
    --train_batch_size=4 \
    --eval_batch_size=4 \
    --eval_samples=56  \
    --num_examples_per_epoch=56  \
    --num_epochs=1  \
    --run_epoch_in_child_process=False \
    --hparams="num_classes=20,moving_average_decay=0,mixed_precision=true"
$ wget https://storage.googleapis.com/cloud-tpu-checkpoints/efficientnet/lite/efficientnet-lite0.tar.gz
python3 model_inspect.py \
--runmode=saved_model \
--model_name=efficientdet-lite0 \
--ckpt_path=efficientnet-lite0 \
--saved_model_dir=savedmodeldir \
--tflite_path=efficientdet-d0.tflite

    'efficientdet-lite0':
        dict(
            name='efficientdet-lite0',
            backbone_name='efficientnet-lite0',
            image_size=320,
            fpn_num_filters=64,
            fpn_cell_repeats=3,
            box_class_repeats=3,
            act_type='relu',
            fpn_weight_method='sum',
            anchor_scale=3.0,
        ),
このスクラップは2021/03/30にクローズされました
ログインするとコメントできます