Thanks for all your awesome work in Spatio-Temporal CNNs! I had a couple quick questions about going from a trained model (.mdl format) to a standalone predictor.
I was able to successfully build the environment and train the R(2+1)D model on my own dataset but now that I've got my trained .mdl file I've been struggling to find documentation on how to go from the trained/training model that takes in LMDB's to a 'predictor' model that can take in plain np array inputs.
I've currently got my .mdl loaded into an init_net and predict_net as suggested by much of the documentation I was able to find:
Hey team,
Thanks for all your awesome work in Spatio-Temporal CNNs! I had a couple quick questions about going from a trained model (.mdl format) to a standalone predictor.
I was able to successfully build the environment and train the R(2+1)D model on my own dataset but now that I've got my trained .mdl file I've been struggling to find documentation on how to go from the trained/training model that takes in LMDB's to a 'predictor' model that can take in plain np array inputs.
I've currently got my .mdl loaded into an init_net and predict_net as suggested by much of the documentation I was able to find:
init_net proto:
first two layers in predict_net proto:
My questions are: a) how would I go about modifying the net(s) to allow for individual predictions from raw np arrays? Ex:
and b) how would I switch from GPU to CPU execution? Is there something like workspace.RunAllOnCPU() for example?
Thanks!