NVlabs / mask-auto-labeler

Other
159 stars 14 forks source link

MAL model deployment #4

Closed d710055071 closed 1 year ago

d710055071 commented 1 year ago

I want to export the MAL model, and then use python for reasoning to get the mask img. What should I do?

voidrank commented 1 year ago

Hi @d710055071 ,

  1. Please kindly read the README and then make your questions more specific.
  2. There will be another open-source solution released next month. I will update the README and point everyone to that repo/lib.

Best,

Shiyi

tianyufang1958 commented 1 year ago

Hi @voidrank

I successfully ran python main.py and implement the inference python main.py --resume work_dirs/coco/last.ckpt --label_dump_path psuedo --not_eval_mask, I got two files psuedo and psuedo.result. I don't know what theses two files are and I don't a resulting folder of psuedo labelled images. Should I expect a folder of labelled images? I can see the annotation data inside the psuedo.result, but why it is not .json. What is the information in psuedo?

voidrank commented 1 year ago

The mask labels should be inside JSON file as far as I remember.

tianyufang1958 commented 1 year ago

The mask labels should be inside JSON file as far as I remember.

Unfortunately it doesn't generate .json for me. I have to add the json extension myself. I am not sure if I did correctly.

voidrank commented 1 year ago

oh I forgot, you need to add .json on your own