Addressed review comments. Made the following main changes
1. Added calculation of batch_dim in NMSLiteShapeFn 2. Added a check for max_total_size to be > 0 3. Updated clip calculation to account for all clip co-ordinates 4. Other code style related changes
Loading
Please sign in to comment