Giters

DIYer22 / bpycv

Computer vision utils for Blender (generate instance annoatation, depth and 6D pose by one line code)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

To cityscape format?

Goooyi opened this issue · comments

commented

Thanks for this work, I notice that bpycv support cityscape format, I want to ask is there a demo or doc to demonstrate how to save the results as cityscape format?

commented

Cityscapes style: Similar to *_instanceIds.png in Cityscapes dataset, those png file are Height * Width * 16bit. Each pixel value x means that the pixel belongs to the instance ID is x.

For instance ID x, the corresponding categories_id is x//1000. For example, instance ID 3002, means it's categories_id id is 3002//1000 = 3

Thus, you should set your object with object["inst_id"] = categories_id * 1000 + index
Then

result = bpycv.render_data()
cv2.imwrite("demo-inst.png", np.uint16(result["inst"]))