I am currently working on applying transfer learning to ssdJacintoNetV2 using the Processor SDK Linux. I have completed initial training, l1 regularization, and sparsification on this model. I had set ssd_size=512x512 in train_image_object_detection.sh before training and all training completed successfully, except test and test_quantize, both of which threw a strange error about memory corruption due to double free.
I set up the environment with environment-setup.sh and ran tidl_model_import.out with a modified configuration file of 738x300 ssdJacintoNetV2. However, when I run the model import tool, I get this error:
=============================== TIDL import - parsing ===============================
Caffe Network File : /home/ubuntu/caffe-jacinto-models/scripts/training/ti-custom-cfg1/JDetNet/20200128_00-56_ds_PSP_dsFac_32_hdDS8_1/sparse/ti-custom-cfg1_ssdJacintoNetV2_iter_120000.caffemodel
Caffe Model File : /home/ubuntu/caffe-jacinto-models/scripts/training/ti-custom-cfg1/JDetNet/20200128_00-56_ds_PSP_dsFac_32_hdDS8_1/sparse/deploy.prototxt
TIDL Network File : /home/ubuntu/tidl_net_jdetNet_ssd_512x512.bin
TIDL Model File : /home/ubuntu/tidl_param_jdetNet_ssd_512x512.bin
Caffe import error: the input layer format is not supported. It must be in an input_shape construct as shown below:
input: "data"
input_shape {
dim: 1
dim: 3
dim: 224
dim: 224
}
I assume that 512x512 input size must be supported as it is an option in the train_image_object_detection.sh script located in caffe-jacinto-models/scripts. Why is it throwing this error and how do I fix it?
Thanks,
Kyle