WongKinYiu / ScaledYOLOv4

Scaled-YOLOv4: Scaling Cross Stage Partial Network
GNU General Public License v3.0
2.01k stars 570 forks source link

小白问题 code在什么地方下载呢? #50

Open litttl opened 3 years ago

litttl commented 3 years ago

nvidia-docker run --name yolov4_csp -it -v your_coco_path/:/coco/ -v your_code_path/:/yolo --shm-size=64g nvcr.io/nvidia/pytorch:20.02-py3 中 your_code_path 这个code需要在什么地方下载呢?

tiny里的 因为我看步骤里有个make

WongKinYiu commented 3 years ago

git clone https://github.com/AlexeyAB/darknet

litttl commented 3 years ago

谢谢大佬

yutao007 commented 3 years ago

难道是在docker 里面使用darknet 吗?这个跟在外面编译darknet 有啥区别吗?即我了解的是docker 里面应该跑的是torch版代码

WongKinYiu commented 3 years ago

pytorch/darknet 都在 docker 裡使用 沒啥差別, docker 佈署比較容易, 也比較不會和其他使用者互相干擾

yutao007 commented 3 years ago

感谢您的答疑。我发现新出scaled_yolov4中yolov4_tiny 的cfg和darknet中的yolov4-tiny 的cfg 是一样的,是不是darknet中训练的效果和新出的scaled_yolov4 中的yolov4_tiny 是一样的?

WongKinYiu commented 3 years ago

yolov4-tiny 是用 darknet 訓練的, 差別只在 train from imagenet pretrained model 和 train from scratch. scaled yolov4 中所有 models 皆為 train from scratch.

yutao007 commented 3 years ago

谢谢您百忙中抽时间答疑。通过您的讲解,和我目前查到的资料,现在理解:下载最新版的darknet就行了(docker版的pytorch/darknet,虽然也能训练,只是为了部署方便,其实对于后期的转换训练后cfg/weights格式加速模型也没有直接帮助),新版darknet中cfg文件中的yolov4_tiny ,yolov4_csp ,yolov4x_mish,分别对应着scaled_yolov4新paper中的yolov4_tiny,yolov4_csp,yolov4_large。 如果我的理解是对的那就使用darknet了,不用去安装docker,然后安装nvidia_docker环境了。

yutao007 commented 3 years ago

我刚才又看了一遍你们的paper,我怎么觉得你们的yolov4_tiny的cfg是不是传错文件了。根据你们的思考,不应该和darknet中的yolov4-tiny的cfg文件一样?