You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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.
The text was updated successfully, but these errors were encountered:
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.
could you provide sample codes for inference(generating segmentation mask from trained DeepMaskAlexNet) or explain how to do it.
Thanks.
The text was updated successfully, but these errors were encountered: