-
Notifications
You must be signed in to change notification settings - Fork 50
Open
Description
I tried to do inference on trained motion segmentation model with motionSegmenter_fullModel.t7.
However I could not find any input loader or sample inference codes.
I only found following piece of inference method code that seems not be able to run.
-- function: inference (used for full scene inference)
function DeepMask:inference()
self:cuda()
utils.linear2convTrunk(self.trunk,self.fSz)
self.trunk:evaluate()
self.trunk:forward(torch.CudaTensor(1,3,800,800))
if self.flow then
utils.linear2convHead(self.flowBranch)
self.flowBranch:evaluate()
self.flowBranch:forward(torch.CudaTensor(1,512,300,300))
return
end
utils.linear2convHead(self.maskBranch.modules[1])
self.maskBranch = self.maskBranch.modules[1]
self.maskBranch:evaluate()
self.maskBranch:forward(torch.CudaTensor(1,512,300,300))
if self.color then
utils.linear2convHead(self.colorBranch)
self.colorBranch:evaluate()
self.colorBranch:forward(torch.CudaTensor(1,512,300,300))
else
utils.linear2convHead(self.scoreBranch)
self.scoreBranch:evaluate()
self.scoreBranch:forward(torch.CudaTensor(1,512,300,300))
end
end
could you provide sample codes for inference(generating segmentation mask from trained DeepMaskAlexNet) or explain how to do it.
Thanks.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels