Tim

執行出現 Legacy autograd function with non-static forward method is deprecated.

2

您好，我在執行train.py的時候出現這個訊息 RuntimeError: Legacy autograd function with non-static forward method is deprecated. Please use new-style autograd function with static forward method. 檢查了一下 好像在trainer.py的 roi_cls_loc, roi_score = self.faster_rcnn.head( features, sample_roi, sample_roi_index) 就停止了。...

請問評估自己的模型時，可以得到各類別(class)自己的Precision嗎?

3

執行eval.py可以正確得到模型的AP結果沒錯，但如果想知道各個類別自己的Precision應該如何修改呢? 因為想知道該去加強哪個類別的訓練資料 謝謝

百度空間的預訓練權重檔案格式是.h5

1

請問大神: 百度空間的預訓練權重檔案格式是.h5，好像不是pytorch通常使用的.pth? 不知道是不是這個原因，導致在載入預訓練模型時跳出『UnpicklingError: invalid load key, 'H'.』的訊息 需要將train.py改成 model_path = ' ' pretrained = False 才能正常運行

請問輸入大小除了544x544還有其他選擇嗎?

1

因為顯存真的不高 想詢問看看是否可以調小點?