search

ptran1203 / traffic_sign_detection

Zalo aI 2020: Traffic sign detection using Retinanet and image tiling
26 stars 9 forks source link

So many epoches #5

Open Dohongha opened 3 years ago

Dohongha commented 3 years ago

The epoches are about 17. They consume a lot of time to run. Do you have any idea to improve this. image

ptran1203 commented 3 years ago

If you are training it from scratch, It would take a long time to have good results. there is a pre-trained weight which you can find at /content/drive/MyDrive/zalo_ai/weight_dense.h5

Replace

WEIGHT_FILE = ZALO_DIR + "weight_dense.h5"

by

WEIGHT_FILE = "/content/drive/MyDrive/zalo_ai/weight_dense.h5"
Dohongha commented 3 years ago

thank you so much!!!