如何解决Tensorflow对象检测API:TensorBoard中损坏的训练图像
我在TensorFlow 2上使用TensorFlow对象检测API,我对TensorBoard中显示的训练图像有问题,如下所示:
即使评估图像看起来正常,所以问题不在于数据序列化。
为了确保它不是通过图像增强来完成的,我只在配置文件中进行水平翻转。查看完整的配置文件:
model {
center_net {
num_classes: 3
feature_extractor {
type: "resnet_v1_50_fpn"
}
image_resizer {
keep_aspect_ratio_resizer {
min_dimension: 512
max_dimension: 512
pad_to_max_dimension: true
}
}
object_detection_task {
task_loss_weight: 1.0
offset_loss_weight: 1.0
scale_loss_weight: 0.1
localization_loss {
l1_localization_loss {
}
}
}
object_center_params {
object_center_loss_weight: 1.0
min_box_overlap_iou: 0.7
max_box_predictions: 100
classification_loss {
penalty_reduced_logistic_focal_loss {
alpha: 2.0
beta: 4.0
}
}
}
}
}
train_config: {
batch_size: 5
num_steps: 10000
data_augmentation_options {
random_horizontal_flip {
}
}
optimizer {
adam_optimizer: {
epsilon: 1e-7 # Match tf.keras.optimizers.Adam's default.
learning_rate: {
cosine_decay_learning_rate {
learning_rate_base: 0.3e-3
total_steps: 10000
warmup_learning_rate: 0.3e-3
warmup_steps: 1000
}
}
}
use_moving_average: false
}
max_number_of_boxes: 100
unpad_groundtruth_tensors: false
fine_tune_checkpoint_version: V2
fine_tune_checkpoint: "C:/ObjectDetection/FaceMaskDetection/Zoo/centernet_resnet50_v1_fpn_512x512_coco17_tpu-8/checkpoint/ckpt-0"
fine_tune_checkpoint_type: "fine_tune"
}
train_input_reader: {
label_map_path: "C:/ObjectDetection/FaceMaskDetection/Dataset/TFRecord/label_map.pbtxt"
tf_record_input_reader {
input_path: "C:/ObjectDetection/FaceMaskDetection/Dataset/TFRecord/train.record"
}
}
eval_config: {
metrics_set: "coco_detection_metrics"
use_moving_averages: false
batch_size: 1;
}
eval_input_reader: {
label_map_path: "C:/ObjectDetection/FaceMaskDetection/Dataset/TFRecord/label_map.pbtxt"
shuffle: false
num_epochs: 1
tf_record_input_reader {
input_path: "C:/ObjectDetection/FaceMaskDetection/Dataset/TFRecord/eval.record"
}
}
我正在使用centernet_resnet50_v1_fpn_512x512_coco17_tpu-8
。奇怪的是,损失和mAP看起来都合理,我认为我无法通过如此糟糕的训练图像得到这些数字。仅仅是一些可视化错误吗?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。