gautamtata / deepplastic Goto Github PK

@gautamtata hi we noticed this great use case for YOLOv5 and recently highlighted your work in a LinkedIn post:
https://www.linkedin.com/posts/ultralytics_ai-under-the-hood-object-detection-model-activity-6819273852980477952-Jt9C

We were wondering if you'd like to help use produce a blog post on ultralytics.com about how Vision AI can help clean the world's oceans. If you are interested please email me at [email protected]. Thank you!

Hi @gautamtata , I have used your yolov5s weights and code in colab notebook, but while running train.py or detect.py, I got the following error
!python train.py --img 416 --batch 10 --epochs 10 --data '/content/data.yaml' --cfg ./models/custom_yolov5s.yaml --weights /content/best.pt --cache
val: data=../data.yaml, weights=['/content/last.pt'], batch_size=32, imgsz=416, conf_thres=0.001, iou_thres=0.6, task=val, device=, workers=8, single_cls=False, augment=False, verbose=False, save_txt=False, save_hybrid=False, save_conf=False, save_json=False, project=runs/val, name=exp, exist_ok=False, half=True, dnn=False
YOLOv5 🚀 v6.0-213-gd5966c9 torch 1.10.0+cu102 CUDA:0 (Tesla T4, 15110MiB)

Traceback (most recent call last):
File "val.py", line 382, in
main(opt)
File "val.py", line 355, in main
run(**vars(opt))
File "/usr/local/lib/python3.7/dist-packages/torch/autograd/grad_mode.py", line 28, in decorate_context
return func(*args, **kwargs)
File "val.py", line 139, in run
model = DetectMultiBackend(weights, device=device, dnn=dnn, data=data)
File "/content/yolov5/models/common.py", line 308, in init
model = attempt_load(weights if isinstance(weights, list) else w, map_location=device)
File "/content/yolov5/models/experimental.py", line 96, in attempt_load
ckpt = torch.load(attempt_download(w), map_location=map_location) # load
File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 600, in load
with _open_zipfile_reader(opened_file) as opened_zipfile:
File "/usr/local/lib/python3.7/dist-packages/torch/serialization.py", line 242, in init
super(_open_zipfile_reader, self).init(torch._C.PyTorchFileReader(name_or_buffer))
RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory

Hello thanks for the project,

urls for badges in readme are incorrect, it does not open your colab page.