PyTorch_YOLOv4
PyTorch_YOLOv4 copied to clipboard
WongKinYiu
PyTorch implementation of YOLOv4
Traceback (most recent call last): File "train.py", line 438, in train(hyp, opt, device, tb_writer) File "train.py", line 252, in train pred = model(imgs) File "/home/yijie/anaconda3/envs/yolov4/lib/python3.8/site-packages/torch/nn/modules/module.py", line 722, in _call_impl result...
You provide the pretrained model(.weights). How can I get the pretrained model (.pt)?
Exact code for converting the current Darknet weight file into a torch pt file and how to load it?
https://github.com/WongKinYiu/PyTorch_YOLOv4/blob/eb5f1663ed0743660b8aa749a43f35f505baa325/train.py#L76-L86 Even if convert the code to pt, a **ckpt key error occurs**.So, I am temporarily replacing pretrained using the `load_darknet_weights` function, so please tell me how to convert it...
I have trained a model using my custom dataset and had precision value of about 0.72 but when I used the saved weights for test script the value jumped to...
How to solve this problem "Given groups=1, weight of size [64, 32, 3, 3], expected input[2, 1, 1, 1] to have 32 channels, but got 1 channels instead" when i...
hello, i found you write the log: **2021-10-15 - support joint detection, instance segmentation, and semantic segmentation. seg-yolo** where can see the seg-yolo ?
Hi,i want to convert model to onnx,but i was unsuccessful! Do you have any good Suggestions? TKS!
https://github.com/AlexeyAB/darknet/blob/master/cfg/yolov4-p6.cfg
Dear Wong, I wanted to train YOLOv4pacsp-x-mish for my custom dataset. Can you please guide me in this aspect. 1. Which pretrained weights file should I use for transfer learning?...
