jason-li-831202 / Vehicle-CV-ADAS

The project can achieve FCWS, LDWS, and LKAS functions solely using only visual sensors. using YOLOv5 / YOLOv5-lite / YOLOv6 / YOLOv7 / YOLOv8 / YOLOv9 / EfficientDet and Ultra-Fast-Lane-Detection-v2 .
GNU General Public License v3.0
162 stars 38 forks source link

convertPytorchToONNX.py #15

Closed Jilin-Cancer closed 9 months ago

Jilin-Cancer commented 9 months ago

Hello author, your project is very interesting, but I encountered some issues. I successfully converted the convertOnnxToTensorRT.py file and yolo model file to trt. However, when running the convertPytorchToONNX.py file, I encountered an issue and did not receive a specific error message. The operation was unexpectedly terminated as follows:

C: \ Users \ 86152 \ anaconda3 \ envs \ lane det \ python. exe C:/Users/86152/Desktop/AI/code/Vehicle CV ADAS master/TrafficLane Detector/convertPytorch ToONNX. py

Model Type: UFLDV2_ TUSIMPLE

C: \ Users \ 86152 \ anaconda3 \ envs \ lane det \ lib \ site packages \ torch vision \ models_ Utils.py: 209: UserWarning: The parameter 'pretrained' is deprecated since 0.13 and may be removed in the future, please use 'weights' instead

"The parameter '{pretrained param}' is deprecated since 0.13 and may be removed in the future,"

C: \ Users \ 86152 \ anaconda3 \ envs \ lane det \ lib \ site packages \ torch vision \ models_ Utils. py: 223: UserWarning: Arguments other than a weight enum or 'None' for 'weights' are deprecated since 0.13 and may be removed in the future The current behavior is equivalent to passing weights=ResNet18_ Weights IMAGENET1K_ V1 You can also use weights=ResNet18_ Weights DEFAULT to get the most up to date weights

Warnings. warn (msg)

Process completed with exit code -1073741819 (0xC000000 5)

Could the author please provide the converted TRT file required for the project model? Thank you very much!