無法轉換yolov3-tiny.weights為yolov3-tiny.h5
執行以下程式
# 下載 tiny-yolov3 的網路權重,並且把權重轉換為 keras 能夠讀取的格式
tiny_yolo_model_path = "model_data/yolov3-tiny.h5"
tiny_yolo_anchor_file = "model_data/tiny_yolo_anchors.txt"
if not os.path.exists(tiny_yolo_model_path):
print("yolov3-tiny weights doesn't exist, downloading...")
os.system("wget https://pjreddie.com/media/files/yolov3-tiny.weights")
print("Converting yolov3-tiny.weights to yolov3-tiny.h5...")
os.system("python convert.py yolov3-tiny.cfg yolov3-tiny.weights %s" % tiny_yolo_model_path)
if os.path.exists(tiny_yolo_model_path):
print("Done!")
else:
print("Strange, model doesn't exist, pleace check")
結果為
yolov3-tiny weights doesn't exist, downloading...
Converting yolov3-tiny.weights to yolov3-tiny.h5...
Strange, model doesn't exist, pleace check
無法轉換成功
回答列表
-
2020/01/30 上午 09:22黃易辰贊同數:0不贊同數:0留言數:3
我遇到的狀況是gpu ram已經滿了, 嘗試重開kernel並在配置gpu ram前跑完這個cell試試
-
2020/01/30 上午 09:49Jeffrey贊同數:0不贊同數:0留言數:2