Conversation
- split functionality - support multiple URLs - reduce output - support new yolo models
| def make_response(self, url, session): | ||
| pieces = urlparse(url) | ||
| if pieces.netloc in ["docs.google.com", "drive.google.com"]: | ||
| return session.get(url, params={'confirm': True}, stream=True, timeout=self.TIMEOUT) |
There was a problem hiding this comment.
For some reason this simple thing helps to download large files from Google Drive (Ubuntu 22, packaged requests). I'd appreciate additional checks performed by other people using different OSes and locations.
|
|
||
| def get_sub(self, ref, arch): | ||
| print('** {}'.format(self.filename)) | ||
| def handle_bad_download(self, filename): |
There was a problem hiding this comment.
This function has been reduced, I hope the logic is similar to the original. Maybe some cornercases are not handled well, but in general I think it is OK.
| filename='onnx/data/output_resnet50_int8.pb'), | ||
| ]), | ||
|
|
||
| # TODO: bad file |
There was a problem hiding this comment.
Archive downloaded from GitHub has different checksum and contains different files, perhaps this model and related test should be reviewed separately.
| filename='yolov4x-mish.weights'), | ||
| Model( | ||
| name='GSOC2016-GOTURN', # https://github.com/opencv/opencv_contrib/issues/941 | ||
| gdrive='1j4UTqVE4EGaUFiK7a5I_CYX7twO9c5br', |
There was a problem hiding this comment.
We detect Google Drive by URL now.
| sha='fcd0fa401c83bf2b29e18239a9c2c989c9b8669d', | ||
| filename='onnx/models/yolov7_not_simplified.onnx'), | ||
| Model( | ||
| name='yolox_s_inf_decoder', |
There was a problem hiding this comment.
New model from another PR.
…multiple URLs - reduce output - support new yolo models
|
How can I put models in dl.opencv.org? |
❗ This PR supersedes #1127 and #1126