training with .pfm extension

[urbansound8K]

Search before asking

• I have searched the YOLOv5 issues and discussions and found no similar questions.

Question

how to train yolov5 with pfm files?

Additional

No response

[spacewalk01]

I found that opencv imread supports pfm files.

You may try adding 'pfm' extention to IMG_FORMATS list.

CODE:

yolov5/utils/dataloaders.py

Line 39 in 5f1000a

IMG_FORMATS = 'bmp', 'dng', 'jpeg', 'jpg', 'mpo', 'png', 'tif', 'tiff', 'webp' # include image suffixes

[spacewalk01]

created a pr : #9230 (comment)

[urbansound8K]

do I need to change the number of channel?
or just train as RGB images no need to change anything while training pfm only????

[spacewalk01]

Tested with this image, it works like 'jpg' image and you don't need to change number of channels. But in case, check your pfm shape as follows:

[spacewalk01]

I think you don't need to change anything beside adding 'pfm' extension to dataloaders.py as I suggested. When you try, let me know if it works or not. Good luck.

[glenn-jocher]

@urbansound8K good news 😃! Your original issue may now be fixed ✅ in PR #9230 by @spacewalk01. To receive this update:

• Git – git pull from within your yolov5/ directory or git clone https://github.com/ultralytics/yolov5 again
• PyTorch Hub – Force-reload model = torch.hub.load('ultralytics/yolov5', 'yolov5s', force_reload=True)
• Notebooks – View updated notebooks
• Docker – sudo docker pull ultralytics/yolov5:latest to update your image

Thank you for spotting this issue and informing us of the problem. Please let us know if this update resolves the issue for you, and feel free to inform us of any other issues you discover or feature requests that come to mind. Happy trainings with YOLOv5 🚀!

[urbansound8K]

cloud you please open it again?

[glenn-jocher]

@spacewalk01 @urbansound8K could you try training again with PFM images after PR #9230? If it doesn't work please submit another PR with a fix, thank you!

[github-actions]

👋 Hello, this issue has been automatically marked as stale because it has not had recent activity. Please note it will be closed if no further activity occurs.

Access additional YOLOv5 🚀 resources:

• Wiki – https://github.com/ultralytics/yolov5/wiki
• Tutorials – https://docs.ultralytics.com/yolov5
• Docs – https://docs.ultralytics.com

Access additional Ultralytics ⚡ resources:

• Ultralytics HUB – https://ultralytics.com/hub
• Vision API – https://ultralytics.com/yolov5
• About Us – https://ultralytics.com/about
• Join Our Team – https://ultralytics.com/work
• Contact Us – https://ultralytics.com/contact

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐!