Task Description

Implement voxel branch head as tf.keras.layers.Layer, as described in https://github.com/facebookresearch/meshrcnn/blob/main/meshrcnn/modeling/roi_heads/voxel_head.py

Write your code and tests in this folder: https://github.com/PurdueDualityLab/tf-models/tree/mesh_rcnn/official/vision/beta/projects/mesh_rcnn/modeling/heads

🚨 Please follow the Google Style Guide 🚨

Before opening a pull request:

• Differentially tested against original implementation ✅
• All test cases are written and passed 👍
• Code is linted ✏️