pytorch/vision

transforms.ToTensor() for numpy float array in the range of [0.0, 255.0]

Open

#546 建立於 2018年7月13日

在 GitHub 查看
 (6 留言) (4 反應) (0 負責人)Python (6,858 fork)batch import
enhancementhelp wantedmodule: transforms

倉庫指標

Star
 (15,050 star)
PR 合併指標
 (平均合併 12天 8小時) (30 天內合併 14 個 PR)

描述

I had come across a debugging scenario where the ToTensor() didn't convert the numpy float array in the range of [0.0, 255.0] to the range [0.0 to 1.0] due to following lines: https://github.com/pytorch/vision/blob/master/torchvision/transforms/functional.py#L50-L53

Basically, this API assumes that all the float arrays will already be in range [0.0 to 1.0]. Do you think we have to change this behavior?

貢獻者指南

研究方向
檢查 torchvision/transforms/functional.py 中的 ToTensor 轉換，修改邏輯以檢測值在 [0,255] 範圍內的浮點數組，並將其縮放到 [0,1]，而不是假定為 [0,1]。
技術棧
python
領域
machine learning
議題類型
錯誤
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
2
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-3 小時
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
清晰
前置要求
PythonPyTorch basics
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
70