search

youngwanLEE / centermask2

[CVPR 2020] CenterMask : Real-time Anchor-Free Instance Segmentation
Other
772 stars 159 forks source link

i am not familiar with detectron2 , how can i use the centermask2 in pure pytorch network defines and weights ? #86

Open leoluopy opened 3 years ago

leoluopy commented 3 years ago

hi @youngwanLEE as the title has described , i am not familiar with the detectron framework . how can i port the centermask network definition into the pytorch pure implementations ? after the porting the inference code would be like below:

model = CenterMask(ptretrain_model_path="xxx.pth")
classes ,scores  dets , masks = model( input_image )

any suggestions ? @youngwanLEE thanks !

cssdcc1997 commented 3 years ago

same situation. i am not familiar with the detectron framework, too. How to rewrite the code?

abhigoku10 commented 1 year ago

@leoluopy @cssdcc1997 any update on this thread i am also intrested to knw