CropLayerImpl(const LayerParams& params) { setParamsFrom(params); startAxis = params.get<int>("axis", 2); const DictValue *paramOffset = params.ptr("offset"); if (paramOffset) { for (int i = 0; i < paramOffset->size(); i++) offset.push_back(paramOffset->get<int>(i)); } }
Ptr<Layer> createLayerFromCaffe<CropLayer>(LayerParams& params) { int start_axis = params.get<int>("axis", 2); DictValue *paramOffset = params.ptr("offset"); std::vector<int> offset; if (paramOffset) { for (int i = 0; i < paramOffset->size(); i++) offset.push_back(paramOffset->get<int>(i)); } return Ptr<Layer>(CropLayer::create(start_axis, offset)); }