Commit Graph

9 Commits (main)

Author SHA1 Message Date
Uminosachi 247d388ad2 Remove torch.jit.script from SAM2Transforms class 2024-08-06 19:02:54 +09:00
Uminosachi e54e285072 Catch exception of initialize_config_dir in hydra 2024-08-05 11:39:47 +09:00
Uminosachi ec5d27cb74 Suppress UserWarning of get_sdpa_settings function 2024-08-02 10:34:00 +09:00
Uminosachi de9f5f5c71 Pass antialias=True to torchvision Resize function 2024-08-02 09:14:36 +09:00
Uminosachi 3ddf311c64 Fix fallback nms to output same value as original 2024-08-01 15:29:52 +09:00
Uminosachi 1ccb9e462c Add fallback nms for environ without batched_nms 2024-08-01 14:34:20 +09:00
Uminosachi 71dc7fe3ff Cast points np.ndarray from float64 to float32 2024-07-31 19:21:00 +09:00
Uminosachi 99e476a88d Remove torch.backends.cuda.sdp_kernel context 2024-07-31 18:30:54 +09:00
Uminosachi 8fec5cb42e Support downloading and mask generation of SAM 2 2024-07-31 16:26:00 +09:00