训练过程中突然意外中断的问题 && 数据加载速度问题

RangiLyu / nanodet

NanoDet-Plus⚡Super fast and lightweight anchor-free object detection model. 🔥Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphone🔥

Apache License 2.0

5.78k stars 1.04k forks source link

Open Stephen-Hao opened 1 year ago

Stephen-Hao commented 1 year ago

作者您好！我使用NanoDet训练80万张左右的数据，在正常训练的第20个epoch报文件丢失的错误，请问您知道如何更改吗？另外加载百万左右级别的数据时，初始的训练要等好久，您有优化方案吗？

Stephen-Hao commented 1 year ago

这个加载等时间有点久，可以优化吗

ForeverPs commented 1 year ago

nanodet/data/dataset/yolo.py change the img = cv2.imread() to h, w = imagesize.get()