Deploy output dimensions

33 views
Skip to first unread message

priyal...@vvdntech.in

unread,
Nov 15, 2018, 6:29:35 AM11/15/18
to Caffe Users

I'm planning to train a model with custom dataset of 70000 images having various height and width. I created the LMDB without resizing. In the train.prototxt i resized the input to 300x300. But i want to perform object detection in a video of 720p(1280x720 size). how should i do this? Do i need to modify something in deploy.prototxt like -

input_shape {
  dim: 1
  dim: 3
  dim: 1280
  dim: 720
}

I don't wish to resize the video. Please help

Przemek D

unread,
Nov 15, 2018, 8:35:24 AM11/15/18
to Caffe Users
Changing the deploy size is the way to go about this. Note that the dimensions are NCHW, so if you're working with 1280p videos you probably want to do:
input_shape {
  dim
: 1
  dim
: 3

  dim
: 720
  dim
: 1280
}
Also note that if you have Crop layers in your network, you will have to recalculate offsets.
Message has been deleted

priyal...@vvdntech.in

unread,
Nov 16, 2018, 1:21:33 AM11/16/18
to Caffe Users
Thanks for the reply Przemek, So i only need to make change in deploy.prototxt, and i dont have to make any changes in train.prototxt, right? 
Reply all
Reply to author
Forward
0 new messages