search

koba-jon / pytorch_cpp

Deep Learning sample programs using PyTorch in C++
MIT License
241 stars 52 forks source link

plan? #15

Open rmx564354093 opened 2 years ago

rmx564354093 commented 2 years ago

Do you have a plan to implement Libtorch version YOLOV4 or YOLOV5 or YOLOX can be trained?

koba-jon commented 2 years ago

Yes, I don't know when that will be, but I plan to implement these. My goal is to implement YOLOv3, v4, v5, X in that order. I hope you'll be looking forward to it.

rmx564354093 commented 2 years ago

I really need the libtorch training version of yolo5 or yoloX, but it’s a bit difficult for me to implement it myself Looking forward to your realization