## 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 :rotating_light: Please follow the [Google Style Guide](https://google.github.io/styleguide/pyguide.html) :rotating_light: ## Before opening a pull request: - [ ] Differentially tested against original implementation :white_check_mark: - [x] All test cases are written and passed :+1: - [x] Code is linted :pencil2:
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: