I am experiencing the same issue too with a custom dataset. I am using the faster_rcnn_inception_resnet_v2_1024x1024_coco17_tpu model from tensorflow model zoo.
Trained under Ubuntu 20.04 using CPU, and the eval results come out okay.
When I trained the data using metal on macOS 12.0.1, the prediction results of the training evaluation are garbage. They don't align with class names either. Tensorboard shows the bounding boxes without class labels, and most bounding boxes are scattered in bottom left corner.
I also reran just the tensorflow eval using Metal on the checkpoints from the Ubuntu CPU training (which gave good results on Ubuntu) and output was also garbage. I used the tensorflow model_main_tf2.py script to train and evaluate in all experiments.
So I can also reproduce your issue @AdkPete. Something isn't right here.
Versions for me are:
tensorflow-macos 2.6.0
tensorflow-metal 0.2.0
tf-models-official 2.6.0
tensorflow-deps 2.6.0
python 3.8.12
macOS 12.0.1
16" M1 Max Macbook Pro
Topic:
Machine Learning & AI
SubTopic:
General
Tags: