Skip to content

Instantly share code, notes, and snippets.

@THEFASHIONGEEK

THEFASHIONGEEK/efficientdet_train.py

Last active February 18, 2020 16:52
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save THEFASHIONGEEK/d4d351abcf67a94676312c9e5277e097 to your computer and use it in GitHub Desktop.
Save THEFASHIONGEEK/d4d351abcf67a94676312c9e5277e097 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
efficientdet_train.py
import os
import sys
sys.path.append("../../4_efficientdet/lib/")
from train_detector import Detector
gtf = Detector()
gtf.Train_Dataset(root_dir, coco_dir, img_dir, set_dir, batch_size=8, image_size=512, use_gpu=True)
gtf.Model()
gtf.Set_Hyperparams(lr=0.0001, val_interval=1, es_min_delta=0.0, es_patience=0)
gtf.Train(num_epochs=30, model_output_dir="trained/")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment