roboflow / roboflow/roboflow-python

Unable to up deploy a custom trained yolov5 model on robotflow

Aperta
#351 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub

Nessuno ha ancora preso questa issue.

Lingua principale
Python
Stelle
629
Fork
140
Merge medio
2g 2h
PR unite (30g)
5

Descrizione

``
It has been impossible to upload a yolov5 model trained on a custom dataset. My folder


rf = roboflow.Roboflow(api_key="..")
project = rf.workspace().project("...")

version = project.version(2)

version.deploy("yolov5", "yolov5/runs/train/exp2")

I've got this error

popi

It' an issue that arise as resulting of calling the function torch.load on the parameter : os.path.join(model_path, filename). This function attempts to load the yolov5 model using the same argument used as input for the function deploy. Recall the functions deploy inputs :

(method) def deploy(
model_type: str,
model_path: str,
filename: str = "weights/best.pt"
) -> None
Uploads provided weights file to Roboflow.

Args:
model_type (str): The type of the model to be deployed.
model_path (str): File path to the model weights to be uploaded.
filename (str, optional): The name of the weights file. Defaults to "weights/best.pt".

I tried :

torch.load(os.path.join("yolov5/runs/train/exp2", "weights/best.pt"))

but still getting the same error. According to my research, loading former yolo model versions has usually been done using torch.hub.load rather than torch.load. In fact

torch.hub.load('.', 'custom', path='/path/to/yolov5/runs/train/exp5/weights/best.pt', source='local')

works fine.

Guida per i contributori

Apri la guida per i contributori

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Direzione di ricerca

Inizia da version.deploy(model_type, model_path, filename) e traccia la chiamata a torch.load per i pesi yolov5. Riproduci il problema con il percorso del modello personalizzato fornito e confrontalo con l’esempio funzionante di torch.hub.load. Il lavoro è completato quando un file di pesi yolov5 addestrato personalizzato può essere caricato correttamente tramite deploy.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Valutazione

Stack tecnologico
python, pytorch
Ambito
computer-vision, machine-learning
Tipo di issue
Bug
Difficoltà
3/5
Tempo stimato
1-2 giorni
Stato di attività
Ferma
Chiarezza
Abbastanza chiara
Idoneità per principianti
38/100

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.