models icon indicating copy to clipboard operation
models copied to clipboard

speedup image classification processing

Open shippingwang opened this issue 6 years ago • 1 comments

Speedup Image Classification training speed


  • [x] add eval shape arguments, for different image shape processing, like inception v4
  • [x] add multiprocess reader, set cv2 as the default reader
  • [x] call reading files func in every training loop, and use pass id as the seed to implement global shuffle
  • [x] run normalization operations onto GPU, use uint8 to transfer data in order to use fewer bytes

Experiments Results:

shippingwang avatar Apr 25 '19 19:04 shippingwang

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
You have signed the CLA already but the status is still pending? Let us recheck it.

CLAassistant avatar Mar 24 '20 02:03 CLAassistant