np.where1 [Pytorch] pytorch 에서 np.where 처럼 index 가져오기 a = np.array([[1, 2, 3, 4], [5, 6, 7, 8]]) np.where (iou == IoU_max_per_object).nonzero() C:\Users\csm81\Desktop\projects_3 (detection)\Faster_RCNN_Pytorch\model\target_builder.py:1: UserWarning: This overload of nonzero is deprecated: nonzero() Consider using one of the following signatures instead: nonzero(*, bool as_tuple) (Triggered internally at ..\torch\csrc\utils\python_arg_parser.cpp:882.. 2022. 8. 17. 이전 1 다음