Skip to content
#

libtorch

Here are 133 public repositories matching this topic...

yolov5-rt-stack
zhiqwang
zhiqwang commented Apr 7, 2022

I used following commad to export the ONNX models, and I use the 5.0 tag of ultralytics/yolov5 to train the model.pt. It raises an AttributeError: conv object has no attribute weight. How can I do for this error?

python tools/export_model.py --checkpoint_path model.pt --size_divisible 32

_Originally posted by @Deronjey in zhiqwang/yolov5-rt-stack#159 (comment)

documentation good first issue

A c++ trainable semantic segmentation library based on libtorch (pytorch c++). Backbone: VGG, ResNet, ResNext. Architecture: FPN, U-Net, PAN, LinkNet, PSPNet, DeepLab-V3, DeepLab-V3+ by now.

  • Updated May 19, 2022
  • C++

Improve this page

Add a description, image, and links to the libtorch topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the libtorch topic, visit your repo's landing page and select "manage topics."

Learn more