dangweili / pedestrian-attribute-recognition-pytorch

A simple baseline for pedestrian attribute recognition in surveillance scenarios
332 stars 80 forks source link

improve the recognition speed #19

Open kclch opened 5 years ago

kclch commented 5 years ago

Hello, after I trained the attacked model, it took a long time to test a picture. I tested a picture on the CPU for 3~4s. Is this normal? How to improve the recognition speed? thank you very much

dangweili commented 5 years ago

@kclch I didn't optimize the speed. You can try some lighten networks, such as MobileNet, for training; Or try some other model compression methods.