YOLOV7 Custom data training failed

Hi, I encountered an error during the initiation of the YOLOV7 training on the custom dataset. Made 80% for Testing and 20% for the training of the dataset. The problem as follows:

Traceback (most recent call last):
  File "train.py", line 21, in <module>
    from torch.utils.tensorboard import SummaryWriter
  File "/home/jetson_test/.local/lib/python3.6/site-packages/torch/utils/tensorboard/__init__.py", line 8, in <module>
    from .writer import FileWriter, SummaryWriter  # noqa F401
  File "/home/jetson_test/.local/lib/python3.6/site-packages/torch/utils/tensorboard/writer.py", line 9, in <module>
    from tensorboard.compat.proto.event_pb2 import SessionLog
  File "/usr/local/lib/python3.6/dist-packages/tensorboard/compat/proto/event_pb2.py", line 17, in <module>
    from tensorboard.compat.proto import summary_pb2 as tensorboard_dot_compat_dot_proto_dot_summary__pb2
  File "/usr/local/lib/python3.6/dist-packages/tensorboard/compat/proto/summary_pb2.py", line 17, in <module>
    from tensorboard.compat.proto import tensor_pb2 as tensorboard_dot_compat_dot_proto_dot_tensor__pb2
  File "/usr/local/lib/python3.6/dist-packages/tensorboard/compat/proto/tensor_pb2.py", line 16, in <module>
    from tensorboard.compat.proto import resource_handle_pb2 as tensorboard_dot_compat_dot_proto_dot_resource__handle__pb2
  File "/usr/local/lib/python3.6/dist-packages/tensorboard/compat/proto/resource_handle_pb2.py", line 16, in <module>
    from tensorboard.compat.proto import tensor_shape_pb2 as tensorboard_dot_compat_dot_proto_dot_tensor__shape__pb2
  File "/usr/local/lib/python3.6/dist-packages/tensorboard/compat/proto/tensor_shape_pb2.py", line 23, in <module>
    serialized_pb=_b('\n+tensorboard/compat/proto/tensor_shape.proto\x12\x0btensorboard\"{\n\x10TensorShapeProto\x12.\n\x03\x64im\x18\x02 \x03(\x0b\x32!.tensorboard.TensorShapeProto.Dim\x12\x14\n\x0cunknown_rank\x18\x03 \x01(\x08\x1a!\n\x03\x44im\x12\x0c\n\x04size\x18\x01 \x01(\x03\x12\x0c\n\x04name\x18\x02 \x01(\tB\x87\x01\n\x18org.tensorflow.frameworkB\x11TensorShapeProtosP\x01ZSgithub.com/tensorflow/tensorflow/tensorflow/go/core/framework/tensor_shape_go_proto\xf8\x01\x01\x62\x06proto3')
TypeError: __new__() got an unexpected keyword argument 'serialized_options'


Command as Follows:

python3 train.py --workers 8 --device 0 --batch-size 32 --data data.yaml --img 640 640 epochs 100 --cfg cfg/training/yolov7x.yaml --weights 'yolov7.pt' --name yolov7x hyp /home/jetson_test/yolov7/data/hyp.scratch.p5.yaml

Thanks in advance.

Hi,

Do you want to try to apply the training directly on the Jetson Nano?
Where the train.py comes from? Do you use jetson_inference?

Thanks.

This train.py is in the yolov7 folder of this repository GitHub - WongKinYiu/yolov7: Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

Hi,

Sorry for the late update.

The error seems to be related to some compatibility issue.
Could you check with the GitHub author to see which TensorFlow/Tensorboard version is required first?

Thanks.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.