yolox-pytorch

yolox-pytorch copied to clipboard

CarryHJR

按照楼主的方法，用自己的数据集训练模型发生报错，发生这样的错误 C:/cb/pytorch_1000000000000/work/aten/src/ATen/native/cuda/Loss.cu:102: block: [11,0,0], thread: [23,0,0] Assertion `input_val >= zero && input_val = zero && input_val = zero && input_val = zero && input_val = zero && input_val = zero...

Arcofcosmos

for batch_i, batch in enumerate(train_loader): inps, targets, img_info, ind = batch

1

想问一下大佬们为什么我在运行dataset.py时会报错 for batch_i, batch in enumerate(train_loader): inps, targets, img_info, ind = batch 。 运行train.py时for iter_id, (inps, targets, img_info, ind) in enumerate(data_loader):这里会报错 。 是因为我有什么前置需要处理的信息没有处理吗？

939889912

进行evaluate时候出现问题

8

进行评估的时候运行sh evaluate.sh的时候训练出来的模型出现如下的问题，RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

Xu-feng-feng

ncnn 部署

7

考虑 ncnn 部署代码的实现么

jolt2017

训练时 cuda 指定问题

3

hi： 支持一波，写的很棒，已经用上了，感谢。发现一个小问题，训练时 改变 train.sh gpus 参数 并不能指定 gpu。

jolt2017

开发了一个darknet 原始训练数据集转coco的工具

2

https://github.com/RapidAI/YOLO2COCO 目前支持darknet训练数据和yolov5数据转成 coco 数据，方便大家训练原有数据。

znsoftm

仓主可以说明一下和原版yolox的差异点吗？

5

在预训练权重帖子中看到需要转换才可以使用yolox预训练权重，那问题来了：本实现和原版的在网络结构上有什么区别？ 这涉及到推理代码的实现，比如能不能直接白嫖原版的推理实现代码？还是需要重新实现？

znsoftm

bug: fixed path of coco 128

3

Traceback (most recent call last): File "train.py", line 238, in main() File "train.py", line 212, in main train_loader, val_loader = get_dataloader(opt, no_aug=no_aug) File "/home/znsoft/yolox-pytorch/data/coco_dataset.py", line 18, in get_dataloader train_dataset =...

znsoftm

I am getting this error after successful execution of the 2nd epoch and when it tries to write the mAP results in results.json file.

AhsanAsghar