亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

運行 yad2k 生成模型 h5 文件時,tensorflow 2.3 出現“tensorflow”

運行 yad2k 生成模型 h5 文件時,tensorflow 2.3 出現“tensorflow”

慕少森 2023-08-22 15:02:56
我正在嘗試生成 YOLOv2 模型 yolo.h5,以便我可以加載這個預先訓練的模型。我正在嘗試將 Andrew Ng coursera Yolo 作業(在tensorflow 1.x 中運行)移植到tensorflow 2.3。由于tensorflow uprade( https://www.tensorflow.org/guide/upgrade),我能夠干凈地移植它,但我幾乎沒有意識到我無法下載 yolo.h5 文件(要么它被損壞,要么下載時間太長)出來),因此我認為我應該構建一個,并按照https://github.com/JudasDie/deeplearning.ai/issues/2的說明進行操作。當我克隆 YAD2k 存儲庫并下載 yolo.weights 和 yolo.cfg 時,它看起來非常簡單。我按照說明運行了以下命令:python yad2k.py yolo.cfg yolo.weights model_data/yolo.h5但我收到以下錯誤:-Traceback (most recent call last):  _main(parser.parse_args())File "yad2k.py", line 233, in _main  Lambda(File "/home/sunny/miniconda3/lib/python3.8/site-packages/tensorflow/python/keras/engine/base_layer.py", line 925, in __call__  return self._functional_construction_call(inputs, args, kwargs,File "/home/sunny/miniconda3/lib/python3.8/site-packages/tensorflow/python/keras/engine/base_layer.py", line 1117, in _functional_construction_call  outputs = call_fn(cast_inputs, *args, **kwargs)File "/home/sunny/miniconda3/lib/python3.8/site-packages/tensorflow/python/keras/layers/core.py", line 903, in call  result = self.function(inputs, **kwargs)File "/home/sunny/YAD2K/yad2k/models/keras_yolo.py", line 32, in space_to_depth_x2  return tf.space_to_depth(x, block_size=2)AttributeError: module 'tensorflow' has no attribute 'space_to_depth'從所有聊天中,我發現上述內容需要在tensorflow 1.x 中運行。然而,它讓我回到了開始的地方,即在 TensorFlow 1.x 中運行它。我愿意堅持使用 TensorFlow 2.3。想知道是否有人可以在這里指導我。坦率地說,要讓我繼續下去,我所需要的只是一個 hd5 模型文件。但我認為生成一個比得到一個更好的學習。
查看完整描述

1 回答

?
慕雪6442864

TA貢獻1812條經驗 獲得超5個贊

當您將 yad2k 存儲庫下的所有代碼(特別是 models 文件夾下的 yad2k.py 和 python 文件)升級到tensorflow 2.x 時,上述問題就會消失。tensorflow 提供的精美升級實用程序通過將原始調用替換為兼容的tf.compat.v1.space_to_depth(input=x, block_size=...)

因此,對于那些計劃進行 TensorFlow 和 Keras 降級的艱苦工作的人,我建議他們嘗試 TensorFlow 升級。這節省了大量時間。

這負責我的模型 h5 文件的創建。我的錯——我問這個問題時沒有考慮到這一點。


查看完整回答
反對 回復 2023-08-22
  • 1 回答
  • 0 關注
  • 3464 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號